fixes wrappers type tests

dlt-hub · Oct 30, 2023 · 7c32205 · 7c32205
1 parent 726ea07
commit 7c32205
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/dlt/extract/wrappers.py b/dlt/extract/wrappers.py
@@ -1,5 +1,6 @@
 from typing import Any
 
+from dlt.common.typing import NoneType
 from dlt.common.exceptions import MissingDependencyException
 
 
@@ -9,7 +10,7 @@
 
     PandaFrame, ArrowTable, ArrowRecords = pandas.DataFrame, pyarrow.Table, pyarrow.RecordBatch
 except MissingDependencyException:
-    PandaFrame, ArrowTable, ArrowRecords = None, None, None
+    PandaFrame, ArrowTable, ArrowRecords = NoneType, NoneType, NoneType
 
 
 def wrap_additional_type(data: Any) -> Any: