diff --git a/nequip/scripts/train.py b/nequip/scripts/train.py index 02460aad..7ba1538b 100644 --- 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,