ruff

spiraldb · Oct 23, 2024 · c09e5a2 · c09e5a2
1 parent c3af36f
commit c09e5a2
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pyvortex/python/vortex/dataset.py b/pyvortex/python/vortex/dataset.py
@@ -132,7 +132,7 @@ def take(
         use_threads: bool | None = None,
         memory_pool: pa.MemoryPool = None,
     ) -> pa.Table:
-        return self._dataset.to_array(columns, batch_size, filter).take(encoding.array(indices)).to_arrow_table()
+        return self._dataset.to_array(columns, filter).take(encoding.array(indices)).to_arrow_table()
 
     def to_record_batch_reader(
         self,
@@ -211,7 +211,7 @@ def to_table(
         del memory_pool
         if filter is not None:
             filter = arrow_to_vortex_expr(filter, self.schema)
-        return self._dataset.to_array(columns, batch_size, filter).to_arrow_table()
+        return self._dataset.to_array(columns, filter).to_arrow_table()
 
 
 class VortexScanner(pa.dataset.Scanner):