use try/except in import_plugins function

flyteorg · Jan 31, 2021 · 60b7068 · 60b7068
1 parent a2d477f
commit 60b7068
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/flytekit/plugins/__init__.py b/flytekit/plugins/__init__.py
@@ -40,4 +40,7 @@ def import_plugins():
     import importlib
 
     for plugin in ["awssagemaker", "hive", "kfpytorch", "kftensorflow", "pandera", "papermill", "pod", "spark"]:
-        importlib.import_module(f"flytekitplugins.{plugin}")
+        try:
+            importlib.import_module(f"flytekitplugins.{plugin}")
+        except ImportError:
+            pass