From 079b1b75ff4759b3b48e83f7350fee60052e7a81 Mon Sep 17 00:00:00 2001
From: ahuang11 <huang.andrew12@gmail.com>
Date: Sun, 18 Feb 2024 15:06:19 -0800
Subject: [PATCH] Remove unnecessary dependencies import

---
 tastymap/__init__.py | 8 ++++++--
 tastymap/ai.py       | 2 +-
 2 files changed, 7 insertions(+), 3 deletions(-)

diff --git a/tastymap/__init__.py b/tastymap/__init__.py
index 2686ce5..8afffb3 100644
--- 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
index c2d1041..32f81c8 100644
--- 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"