From edff34172d1efcf7e06f3712ba4325b1dc3af89f Mon Sep 17 00:00:00 2001 From: Jay Chia Date: Fri, 20 Dec 2024 23:04:15 +0800 Subject: [PATCH] Comments --- daft/dataframe/dataframe.py | 9 ++------- 1 file changed, 2 insertions(+), 7 deletions(-) diff --git a/daft/dataframe/dataframe.py b/daft/dataframe/dataframe.py index 10f79a018b..7174ef73f4 100644 --- a/daft/dataframe/dataframe.py +++ b/daft/dataframe/dataframe.py @@ -870,17 +870,12 @@ def write_deltalake( from packaging.version import parse from daft import from_pydict + from daft.dependencies import unity_catalog from daft.filesystem import get_protocol_from_path from daft.io import DataCatalogTable from daft.io._deltalake import large_dtypes_kwargs from daft.io.object_store_options import io_config_to_storage_options - _UNITY_AVAILABLE = True - try: - from daft.unity_catalog import UnityCatalogTable - except ImportError: - _UNITY_AVAILABLE = False - if schema_mode == "merge": raise ValueError("Schema mode' merge' is not currently supported for write_deltalake.") @@ -903,7 +898,7 @@ def write_deltalake( table_uri = table elif isinstance(table, pathlib.Path): table_uri = str(table) - elif _UNITY_AVAILABLE and isinstance(table, UnityCatalogTable): + elif unity_catalog.module_available() and isinstance(table, unity_catalog.UnityCatalogTable): table_uri = table.table_uri io_config = table.io_config elif isinstance(table, DataCatalogTable):