Skip to content

Commit 91c7396

Browse files
author
Jean-Baptiste Oger
committed
fix: use model_id instead of self.model_id when output_model_name is None in train_model
1 parent c58e75b commit 91c7396

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

opensearch_py_ml/ml_models/sentencetransformermodel.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -421,7 +421,7 @@ def train_model(
421421
if model_id is None:
422422
model_id = self.model_id
423423
if output_model_name is None:
424-
output_model_name = str(self.model_id.split("/")[-1] + ".pt")
424+
output_model_name = str(model_id.split("/")[-1] + ".pt")
425425

426426
# declare variables before assignment for training
427427
corp_len = []

0 commit comments

Comments
 (0)