From 23c1a8664c15c1c0e34504e2a5947a5ac751e04d Mon Sep 17 00:00:00 2001 From: lixfz Date: Fri, 15 Mar 2024 15:01:27 +0800 Subject: [PATCH] Fix pgmpy import err --- ylearn/causal_discovery/__init__.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/ylearn/causal_discovery/__init__.py b/ylearn/causal_discovery/__init__.py index e009200..9b98c01 100644 --- 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)