expose ml_relax in relaxation

FAIR-Chem · Sep 20, 2024 · efcb0af · efcb0af
1 parent 7c6244f
commit efcb0af
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/fairchem/core/common/relaxation/__init__.py b/src/fairchem/core/common/relaxation/__init__.py
@@ -7,6 +7,7 @@
 
 from __future__ import annotations
 
+from .ml_relaxation import ml_relax
 from .optimizers.optimizable import OptimizableBatch, OptimizableUnitCellBatch
 
-__all__ = ["OptimizableBatch", "OptimizableUnitCellBatch"]
+__all__ = ["ml_relax", "OptimizableBatch", "OptimizableUnitCellBatch"]