fix: use model_id instead of self.model_id when output_model_name is …

…None in train_model
opensearch-project · Mar 5, 2024 · 91c7396 · 91c7396
1 parent c58e75b
commit 91c7396
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/opensearch_py_ml/ml_models/sentencetransformermodel.py b/opensearch_py_ml/ml_models/sentencetransformermodel.py
@@ -421,7 +421,7 @@ def train_model(
         if model_id is None:
             model_id = self.model_id
         if output_model_name is None:
-            output_model_name = str(self.model_id.split("/")[-1] + ".pt")
+            output_model_name = str(model_id.split("/")[-1] + ".pt")
 
         # declare variables before assignment for training
         corp_len = []