Searched refs:wrapper_class (Results 1 – 4 of 4) sorted by relevance
79 wrapper_class: Type[DataChunk] = DataChunk,86 self.wrapper_class = wrapper_class93 yield self.wrapper_class(batch)97 yield self.wrapper_class(batch)236 self.wrapper_class = DataChunk277 result: DataChunk[Any] = self.wrapper_class(self.buffer_elements[key])285 result = self.wrapper_class(result_to_yield)289 result = self.wrapper_class(self.buffer_elements.pop(key))306 self.wrapper_class,323 self.wrapper_class,
45 wrapper_class: Type[DataChunk] = DataChunk,52 self.wrapper_class = wrapper_class60 return self.wrapper_class(batch)63 return self.wrapper_class(batch)
79 py::object wrapper_class = in bind_cpp_module_wrapper() local84 wrapper_class.attr("__init__") = py::cpp_function( in bind_cpp_module_wrapper()89 py::is_method(wrapper_class)); in bind_cpp_module_wrapper()93 module.attr(name) = wrapper_class; in bind_cpp_module_wrapper()
399 def batch(self, batch_size=10, drop_last: bool = False, wrapper_class=DataChunkDF): argument401 dp = dp.as_datapipe().batch(1, drop_last=drop_last, wrapper_class=wrapper_class)