Update model_autotracing.py

Signed-off-by: Thanawan Atchariyachanvanit <[email protected]>
thanawan-atc · Jul 19, 2023 · 5a971ef · 5a971ef
1 parent 53a626e
commit 5a971ef
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/utils/model_uploader/model_autotracing.py b/utils/model_uploader/model_autotracing.py
@@ -29,8 +29,7 @@
 
 from opensearch_py_ml.ml_commons import MLCommonClient
 from opensearch_py_ml.ml_models.sentencetransformermodel import SentenceTransformerModel
-import utils
-from utils.model_uploader import OPENSEARCH_TEST_CLIENT
+from model_uploader import OPENSEARCH_TEST_CLIENT
 
 BOTH_FORMAT = "BOTH"
 TORCH_SCRIPT_FORMAT = "TORCH_SCRIPT"