fix olive not found

vladmandic · Oct 31, 2023 · 2fd8d2c · 2fd8d2c
1 parent 7d7821d
commit 2fd8d2c
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/installer.py b/installer.py
@@ -754,6 +754,14 @@ def ensure_base_requirements():
         import rich # pylint: disable=unused-import
     except ImportError:
         pass
+    try: # related to: https://github.com/microsoft/Olive/issues/675
+        import olive.workflows # pylint: disable=unused-import
+    except ImportError:
+        install('olive-ai', 'Olive')
+        try:
+            import olive.workflows
+        except ImportError:
+            log.error('Failed to install dependency: olive-ai.')
 
 
 def install_requirements():

diff --git a/modules/paths.py b/modules/paths.py
@@ -3,7 +3,6 @@
 import sys
 import json
 import argparse
-import olive.workflows
 from modules.errors import log
 
 # parse args, parse again after we have the data-dir and early-read the config file