Remove unnecessary dependencies import

ahuang11 · Feb 18, 2024 · 079b1b7 · 079b1b7
1 parent f7b841a
commit 079b1b7
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 3 deletions.
diff --git a/tastymap/__init__.py b/tastymap/__init__.py
@@ -2,8 +2,12 @@
 
 from .core import cook_tmap, pair_tbar
 from .models import TastyBar, TastyMap
-from .ui import TastyKitchen
 
-__version__ = "0.4.0"
+try:
+    from .ui import TastyKitchen
+except ImportError:
+    pass
+
+__version__ = "0.4.1"
 
 __all__ = ["cook_tmap", "pair_tbar", "TastyMap", "TastyBar", "TastyKitchen"]
diff --git a/tastymap/ai.py b/tastymap/ai.py
@@ -1,7 +1,7 @@
 try:
     from marvin import ai_fn, ai_model  # type: ignore
     from pydantic import BaseModel, Field  # type: ignore
-except ImportError:
+except ImportError as exc:
     raise ImportError(
         "Please install marvin and pydantic to use this module, "
         "e.g. pip install marvin pydantic"