Revert "Avoid unnecessary e3nn-related JIT warning" (be22479), fixed …

…upstream by @Linux-cpp-lisp in e3nn/e3nn#437
mir-group · Jun 27, 2024 · 7e1ff56 · 7e1ff56
1 parent b591aa2
commit 7e1ff56
Showing 1 changed file with 0 additions and 4 deletions.
diff --git a/nequip/scripts/train.py b/nequip/scripts/train.py
@@ -24,10 +24,6 @@
 from nequip.utils._global_options import _set_global_options
 from nequip.scripts._logger import set_up_script_logger
 
-warnings.filterwarnings(  # unnecessary e3nn-related JIT warning
-    "ignore",
-    message="The TorchScript type system doesn't support instance-level annotations",
-)
 default_config = dict(
     root="./",
     tensorboard=False,