From 7c32205e8657768fc270effd4490a48aaea48b57 Mon Sep 17 00:00:00 2001 From: Marcin Rudolf Date: Mon, 30 Oct 2023 12:54:34 +0100 Subject: [PATCH] fixes wrappers type tests --- dlt/extract/wrappers.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/dlt/extract/wrappers.py b/dlt/extract/wrappers.py index 500cd5f78e..e8e295f245 100644 --- 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: