From efcb0af2eac41dfb42e081e79aa5e3785dd63ebb Mon Sep 17 00:00:00 2001 From: lbluque Date: Fri, 20 Sep 2024 11:03:18 -0700 Subject: [PATCH] expose ml_relax in relaxation --- src/fairchem/core/common/relaxation/__init__.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/fairchem/core/common/relaxation/__init__.py b/src/fairchem/core/common/relaxation/__init__.py index 9fea399f0..068eaf718 100644 --- 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"]