remove old imports

FAIR-Chem · Jul 18, 2023 · 4961bb1 · 4961bb1
1 parent 8bac184
commit 4961bb1
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 5 deletions.
diff --git a/ocpmodels/tasks/task.py b/ocpmodels/tasks/task.py
@@ -9,7 +9,6 @@
 import os
 
 from ocpmodels.common.registry import registry
-from ocpmodels.trainers.forces_trainer import ForcesTrainer
 
 
 class BaseTask:

diff --git a/ocpmodels/trainers/__init__.py b/ocpmodels/trainers/__init__.py
@@ -5,10 +5,8 @@
 
 __all__ = [
     "BaseTrainer",
-    "ForcesTrainer",
-    "EnergyTrainer",
+    "OCPTrainer",
 ]
 
 from .base_trainer import BaseTrainer
-from .energy_trainer import EnergyTrainer
-from .forces_trainer import ForcesTrainer
+from .ocp_trainer import OCPTrainer