| 38 | } |
| 39 | |
| 40 | auto validate_argnums_argnames( |
| 41 | const std::optional<IntOrVec>& argnums, |
| 42 | const StrOrSet& argnames) { |
| 43 | std::unordered_set<std::string> setnames; |
| 44 | if (auto pv = std::get_if<std::string>(&argnames); pv) { |
| 45 | setnames = {*pv}; |
| 46 | } else { |
| 47 | setnames = std::get<std::unordered_set<std::string>>(argnames); |
| 48 | } |
| 49 | |
| 50 | if (!argnums.has_value()) { |
| 51 | // argnums was not provided and argnames was empty |
| 52 | if (setnames.empty()) { |
| 53 | return std::make_pair(std::vector<int>{0}, setnames); |
| 54 | } else { |
| 55 | return std::make_pair(std::vector<int>{}, setnames); |
| 56 | } |
| 57 | } |
| 58 | |
| 59 | std::vector<int> vecnums; |
| 60 | if (auto pv = std::get_if<int>(&(*argnums)); pv) { |
| 61 | vecnums = {*pv}; |
| 62 | } else { |
| 63 | vecnums = std::get<std::vector<int>>(*argnums); |
| 64 | } |
| 65 | |
| 66 | return std::make_pair(vecnums, setnames); |
| 67 | } |
| 68 | |
| 69 | auto py_value_and_grad( |
| 70 | const nb::callable& fun, |
no test coverage detected