Class PyBufferIterator

python/src/wrap_buffer.cpp:14–33

namespace {
class PyBufferIterator {
 public:
  PyBufferIterator(Buffer& buffer)
      : buffer_(buffer), iter_(0), size_(buffer.size()) {};
  py::dict next() {
    if (iter_ >= size_) {
      throw py::stop_iteration();
    } else {
      Sample res;
      {
        py::gil_scoped_release release;
        res = buffer_.get(iter_++);
      }
      return mlx::pybind::to_py_sample(res);
    }
  };
  Buffer buffer_;
  int64_t iter_;
  int64_t size_;
};
} // namespace
namespace py = pybind11;
using namespace mlx::data;