diff --git a/utils/model_uploader/model_autotracing.py b/utils/model_uploader/model_autotracing.py index 4b312c8ec..5513b5377 100644 --- a/utils/model_uploader/model_autotracing.py +++ b/utils/model_uploader/model_autotracing.py @@ -27,10 +27,13 @@ # sys.path.append(ROOT_DIR) # sys.path.append(TEST_DIR) LICENSE_PATH = "LICENSE" - +#sys.path.append(os.path.abspath(os.path.join("."))) +#print(sys.path) from opensearch_py_ml.ml_commons import MLCommonClient from opensearch_py_ml.ml_models.sentencetransformermodel import SentenceTransformerModel -from opensearch_tests import OPENSEARCH_TEST_CLIENT + +#import opensearch_tests +#from opensearch_tests import OPENSEARCH_TEST_CLIENT BOTH_FORMAT = "BOTH" TORCH_SCRIPT_FORMAT = "TORCH_SCRIPT"