Skip to content

Commit

Permalink
Remove sys.path.append in model_autotracing.py
Browse files Browse the repository at this point in the history
Signed-off-by: Thanawan Atchariyachanvanit <[email protected]>
  • Loading branch information
thanawan-atc authored Jul 19, 2023
1 parent 451bf5b commit 4f1f075
Showing 1 changed file with 6 additions and 6 deletions.
12 changes: 6 additions & 6 deletions utils/model_uploader/model_autotracing.py
Original file line number Diff line number Diff line change
Expand Up @@ -21,15 +21,15 @@
from numpy.typing import DTypeLike
from sentence_transformers import SentenceTransformer

ROOT_DIR = os.path.abspath(os.path.join("../opensearch-py-ml"))
TEST_DIR = os.path.join(ROOT_DIR, "tests")
LICENSE_PATH = os.path.join(ROOT_DIR, "LICENSE")
sys.path.append(ROOT_DIR)
sys.path.append(TEST_DIR)
# ROOT_DIR = os.path.abspath(os.path.join("../opensearch-py-ml"))
# TEST_DIR = os.path.join(ROOT_DIR, "tests")
LICENSE_PATH = "LICENSE"
# sys.path.append(ROOT_DIR)
# sys.path.append(TEST_DIR)

from opensearch_py_ml.ml_commons import MLCommonClient
from opensearch_py_ml.ml_models.sentencetransformermodel import SentenceTransformerModel
from tests import OPENSEARCH_TEST_CLIENT
from utils.model_uploader import OPENSEARCH_TEST_CLIENT

BOTH_FORMAT = "BOTH"
TORCH_SCRIPT_FORMAT = "TORCH_SCRIPT"
Expand Down

0 comments on commit 4f1f075

Please sign in to comment.