add DenoisingForcesTrainer to init of trainers directory

IliasChair · Oct 5, 2024 · 939f6ad · 939f6ad
1 parent 38a47be
commit 939f6ad
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/fairchem/core/trainers/__init__.py b/src/fairchem/core/trainers/__init__.py
@@ -6,8 +6,10 @@
 
 from .base_trainer import BaseTrainer
 from .ocp_trainer import OCPTrainer
+from .denoising_forces_trainer import DenoisingForcesTrainer
 
 __all__ = [
     "BaseTrainer",
     "OCPTrainer",
+    "DenoisingForcesTrainer"
 ]