Print warning message instead of error when dPIE library are not avai…

…lable
Herculens · Nov 6, 2024 · 31def60 · 31def60
1 parent a9643e9
commit 31def60
Showing 1 changed file with 12 additions and 6 deletions.
diff --git a/herculens/MassModel/Profiles/dpie.py b/herculens/MassModel/Profiles/dpie.py
@@ -13,18 +13,24 @@
 
 from herculens.Util import util, param_util
 
-try:
-    from herculens.MassModel.Profiles.glee.piemd_jax import Piemd_GPU
-except ImportError:
-    raise ImportError("Please contact the author to use this dPIE profile "
-                        "as it depends on non-public libraries.")
-
 
 __all__ = [
     'DPIE_GLEE', 
     'DPIE_GLEE_STATIC',
 ]
 
+try:
+    from herculens.MassModel.Profiles.glee.piemd_jax import Piemd_GPU
+except ImportError:
+    print("WARNING: dPIE profile class cannot be loaded. "
+          "If you wish to use this profile, "
+          "please contact the author to use this dPIE profile "
+          "as it depends on non-public libraries.")
+    class DPIE_GLEE(object):
+        pass
+    class DPIE_GLEE_STATIC(object):
+        pass
+
 
 class DPIE_GLEE(object):
     """