| 609 | args_structure_(std::move(args_structure)), |
| 610 | output_structure_(output_structure) {} |
| 611 | ~InnerFunction() { |
| 612 | nb::gil_scoped_acquire gil; |
| 613 | |
| 614 | fun_.reset(); |
| 615 | args_structure_.reset(); |
| 616 | } |
| 617 | |
| 618 | std::vector<mx::array> operator()(const std::vector<mx::array>& inputs) { |
| 619 | auto args = nb::cast<nb::tuple>( |