perf: lazily import pyiceberg and unity catalog if available

daft/catalog/__init__.py

@@ -45,21 +45,12 @@
 
 from daft.dataframe import DataFrame
 
-_PYICEBERG_AVAILABLE = False
-try:
-    from pyiceberg.catalog import Catalog as PyIcebergCatalog
-
-    _PYICEBERG_AVAILABLE = True
-except ImportError:
-    pass
+from typing import TYPE_CHECKING
 
-_UNITY_AVAILABLE = False
-try:
+if TYPE_CHECKING:
+    from pyiceberg.catalog import Catalog as PyIcebergCatalog
     from daft.unity_catalog import UnityCatalog
 
-    _UNITY_AVAILABLE = True
-except ImportError:
-    pass
 
 __all__ = [
     "read_table",
@@ -136,6 +127,22 @@ def register_python_catalog(catalog: PyIcebergCatalog | UnityCatalog, name: str
         >>> daft.catalog.register_python_catalog(catalog, "my_daft_catalog")
 
     """
+    _PYICEBERG_AVAILABLE = False
+    try:
+        from pyiceberg.catalog import Catalog as PyIcebergCatalog
+
+        _PYICEBERG_AVAILABLE = True
+    except ImportError:
+        pass
+
+    _UNITY_AVAILABLE = False
+    try:
+        from daft.unity_catalog import UnityCatalog
+
+        _UNITY_AVAILABLE = True
+    except ImportError:
+        pass
+
     python_catalog: PyIcebergCatalog
     if _PYICEBERG_AVAILABLE and isinstance(catalog, PyIcebergCatalog):
         from daft.catalog.pyiceberg import PyIcebergCatalogAdaptor