fix

Eventual-Inc · Oct 28, 2024 · 0a9f573 · 0a9f573
1 parent 3077a17
commit 0a9f573
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 6 deletions.
diff --git a/daft/daft/__init__.pyi b/daft/daft/__init__.pyi
@@ -751,8 +751,7 @@ class ScanTask:
     def python_factory_func_scan_task(
         func_args: tuple[Any, ...],
         schema: PySchema,
-        module: str | None,
-        func_name: str | None,
+        module_and_func_name: tuple[str, str] | None,
         func: Callable[[int, Any], Iterator[PyTable]] | None,
         num_rows: int | None,
         size_bytes: int | None,

diff --git a/daft/io/_generated.py b/daft/io/_generated.py
@@ -80,8 +80,7 @@ def to_scan_tasks(self, pushdowns: Pushdowns) -> Iterator[ScanTask]:
             yield ScanTask.python_factory_func_scan_task(
                 func_args=(i, *self._generator_args),
                 schema=self.schema()._schema,
-                module=None,
-                func_name=None,
+                module_and_func_name=None,
                 func=self._generator,
                 num_rows=None,
                 size_bytes=None,

diff --git a/daft/io/_lance.py b/daft/io/_lance.py
@@ -124,8 +124,10 @@ def to_scan_tasks(self, pushdowns: Pushdowns) -> Iterator[ScanTask]:
             yield ScanTask.python_factory_func_scan_task(
                 func_args=(fragment, required_columns),
                 schema=self.schema()._schema,
-                module=_lancedb_table_factory_function.__module__,
-                func_name=_lancedb_table_factory_function.__name__,
+                module_and_func_name=(
+                    _lancedb_table_factory_function.__module__,
+                    _lancedb_table_factory_function.__name__,
+                ),
                 func=None,
                 num_rows=num_rows,
                 size_bytes=size_bytes,