Method next

python/src/wrap_stream.cpp:42–60

  }

  virtual Sample next() const {
    Sample sample;
    {
      std::unique_lock lock(mutex_);
      py::gil_scoped_acquire gil;
      try {
        sample = mlx::pybind::to_sample(next_().cast<py::dict>());
      } catch (py::error_already_set& e) {
        if (e.matches(PyExc_StopIteration)) {
          // do nothing
        } else {
          throw;
        }
      } catch (...) {
        throw;
      }
    }
    return sample;
  }

 private:
  py::function iterable_factory_;