remove non compatible typing hint

ACEsuit · Apr 30, 2024 · 7697261 · 7697261
1 parent d690063
commit 7697261
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/mace/cli/fine_tuning_select.py b/mace/cli/fine_tuning_select.py
@@ -4,7 +4,7 @@
 
 import argparse
 import logging
-import typing as t
+from typing import List
 
 
 import ase.data
@@ -96,7 +96,7 @@ def parse_args() -> argparse.Namespace:
 
 
 def calculate_descriptors(
-    atoms: t.List[ase.Atoms | ase.Atom], calc: MACECalculator, cutoffs: None | dict
+    atoms: List[ase.Atoms], calc: MACECalculator, cutoffs: None | dict
 ) -> None:
     logging.info("Calculating descriptors")
     for mol in atoms: