FAIR-Chem · abhshkdz · May 6, 2023 · May 3, 2023 · May 3, 2023
diff --git a/ocpmodels/models/escn/__init__.py b/ocpmodels/models/escn/__init__.py
diff --git a/ocpmodels/models/gemnet/__init__.py b/ocpmodels/models/gemnet/__init__.py
diff --git a/ocpmodels/models/gemnet/layers/__init__.py b/ocpmodels/models/gemnet/layers/__init__.py
diff --git a/ocpmodels/models/gemnet_gp/__init__.py b/ocpmodels/models/gemnet_gp/__init__.py
diff --git a/ocpmodels/models/gemnet_gp/layers/__init__.py b/ocpmodels/models/gemnet_gp/layers/__init__.py
diff --git a/ocpmodels/models/gemnet_oc/__init__.py b/ocpmodels/models/gemnet_oc/__init__.py
diff --git a/ocpmodels/models/gemnet_oc/layers/__init__.py b/ocpmodels/models/gemnet_oc/layers/__init__.py
diff --git a/ocpmodels/models/painn/__init__.py b/ocpmodels/models/painn/__init__.py
diff --git a/ocpmodels/models/scn/__init__.py b/ocpmodels/models/scn/__init__.py
diff --git a/ocpmodels/models/scn/spherical_harmonics.py b/ocpmodels/models/scn/spherical_harmonics.py
@@ -14,13 +14,16 @@
 try:
     from e3nn import o3
     from e3nn.o3 import FromS2Grid, ToS2Grid
-except ImportError:
-    pass
 
-# Borrowed from e3nn @ 0.4.0:
-# https://github.com/e3nn/e3nn/blob/0.4.0/e3nn/o3/_wigner.py#L10
-# _Jd is a list of tensors of shape (2l+1, 2l+1)
-_Jd = torch.load(os.path.join(os.path.dirname(__file__), "Jd.pt"))
+    # Borrowed from e3nn @ 0.4.0:
+    # https://github.com/e3nn/e3nn/blob/0.4.0/e3nn/o3/_wigner.py#L10
+    # _Jd is a list of tensors of shape (2l+1, 2l+1)
+    _Jd = torch.load(os.path.join(os.path.dirname(__file__), "Jd.pt"))
+except (ImportError, FileNotFoundError):
+    logging.error(
+        "Invalid setup for SCN. Either the e3nn library or Jd.pt is missing."
+    )
+    pass
 
 
 class SphericalHarmonicsHelper: