Remove import from atomate2

materialsproject · Apr 10, 2024 · 59580e5 · 59580e5
1 parent a7ffcdd
commit 59580e5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/emmet-core/emmet/core/classical_md/openmm/tasks.py b/emmet-core/emmet/core/classical_md/openmm/tasks.py
@@ -9,7 +9,7 @@
 from emmet.core.vasp.task_valid import TaskState  # type: ignore[import-untyped]
 from pydantic import BaseModel, Field
 
-from atomate2.classical_md.schemas import ClassicalMDTaskDocument
+from emmet.core.classical_md import ClassicalMDTaskDocument
 
 
 class CalculationInput(BaseModel, extra="allow"):  # type: ignore[call-arg]