Fix pgmpy import err

DataCanvasIO · Mar 15, 2024 · 23c1a86 · 23c1a86
1 parent 10c1a81
commit 23c1a86
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/ylearn/causal_discovery/__init__.py b/ylearn/causal_discovery/__init__.py
@@ -36,3 +36,10 @@ def __init__(self, *args, **kwargs):
     class PgmProxy(BaseDiscovery):
         def __init__(self, *args, **kwargs):
             raise ImportError(_msg_pgm)
+except SyntaxError as e:
+    _msg_pgm = f'{e}, re-install pgmpy and try again.'
+
+
+    class PgmProxy(BaseDiscovery):
+        def __init__(self, *args, **kwargs):
+            raise SyntaxError(_msg_pgm)