fix tests

Signed-off-by: kalyan <[email protected]>
rawwar · Nov 3, 2023 · 45d7aeb · 45d7aeb
1 parent a8306fe
commit 45d7aeb
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 4 deletions.
diff --git a/noxfile.py b/noxfile.py
@@ -101,7 +101,7 @@ def lint(session):
     #     if errors:
     #         session.error("\n" + "\n".join(sorted(set(errors))))
 
-
+@nox.session(reuse_venv=True)
 @nox.session(python=["3.8", "3.9", "3.10"])
 @nox.parametrize("pandas_version", ["1.5.0"])
 def test(session, pandas_version: str):

diff --git a/opensearch_py_ml/ml_commons/__init__.py b/opensearch_py_ml/ml_commons/__init__.py
@@ -9,7 +9,6 @@
 
 from opensearch_py_ml.ml_commons.ml_commons_client import MLCommonClient
 from opensearch_py_ml.ml_commons.model_execute import ModelExecute
-from opensearch_py_ml.ml_commons.model_train import ModelTrain
 from opensearch_py_ml.ml_commons.model_uploader import ModelUploader
 
-__all__ = ["MLCommonClient", "ModelExecute", "ModelUploader", "ModelTrain"]
+__all__ = ["MLCommonClient", "ModelExecute", "ModelUploader"]
diff --git a/opensearch_py_ml/ml_commons/ml_commons_client.py b/opensearch_py_ml/ml_commons/ml_commons_client.py
@@ -22,7 +22,6 @@
     TIMEOUT,
 )
 from opensearch_py_ml.ml_commons.model_execute import ModelExecute
-from opensearch_py_ml.ml_commons.model_train import ModelTrain
 from opensearch_py_ml.ml_commons.model_uploader import ModelUploader