improve trajectory type hint

zadorlab · Jun 15, 2023 · 44e97c2 · 44e97c2
1 parent 56b974e
commit 44e97c2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/sella/optimize/optimize.py b/sella/optimize/optimize.py
@@ -42,7 +42,7 @@ def __init__(
         atoms: Atoms,
         restart: bool = None,
         logfile: str = '-',
-        trajectory: str = None,
+        trajectory: Union[str, Trajectory] = None,
         master: bool = None,
         force_consistent: bool = False,
         delta0: float = None,