Skip to content
Snippets Groups Projects
Commit 6f2b9d0b authored by Olivier BICHLER's avatar Olivier BICHLER
Browse files

Hotfix for DataProvider default argument binding

parent 0b6d4f66
No related branches found
No related tags found
1 merge request!212Version 0.3.0
Pipeline #54140 passed
...@@ -27,7 +27,7 @@ std::vector<std::shared_ptr<Aidge::Tensor>> DataProvider::next() { ...@@ -27,7 +27,7 @@ std::vector<std::shared_ptr<Aidge::Tensor>> DataProvider::next() {
void init_DataProvider(py::module& m){ void init_DataProvider(py::module& m){
py::class_<DataProvider, std::shared_ptr<DataProvider>>(m, "DataProvider") py::class_<DataProvider, std::shared_ptr<DataProvider>>(m, "DataProvider")
.def(py::init<Database&, std::size_t, std::string, bool, bool>(), py::arg("database"), py::arg("batch_size"), py::arg("backend"), py::arg("shuffle"), py::arg("drop_last")) .def(py::init<Database&, std::size_t, std::string, bool, bool>(), py::arg("database"), py::arg("batch_size"), py::arg("backend") = "cpu", py::arg("shuffle") = false, py::arg("drop_last") = false)
.def("__iter__", &DataProvider::iter) .def("__iter__", &DataProvider::iter)
.def("__next__", &DataProvider::next) .def("__next__", &DataProvider::next)
.def("__len__", &DataProvider::getNbBatch); .def("__len__", &DataProvider::getNbBatch);
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment