Better error handling for import error

MilesCranmer · Feb 3, 2022 · 5d9233a · 5d9233a
1 parent b1eaa8d
commit 5d9233a
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/pysr/sr.py b/pysr/sr.py
@@ -1050,13 +1050,14 @@ def _run(self, X, y, weights, variable_names):
             Main.eval(
                 f'Pkg.activate("{_escape_filename(self.julia_project)}", io={io})'
             )
+            from julia.api import JuliaError
             try:
                 if update:
                     Main.eval(f"Pkg.resolve(io={io})")
                     Main.eval(f"Pkg.instantiate(io={io})")
                 else:
                     Main.eval(f"Pkg.instantiate(io={io})")
-            except RuntimeError as e:
+            except (JuliaError, RuntimeError) as e:
                 raise ImportError(import_error_string(self.julia_project)) from e
             Main.eval("using SymbolicRegression")