Skip to content

Commit fc09ad1

Browse files
committed
function_name -> model_mask_type
Signed-off-by: HenryL27 <[email protected]>
1 parent 210e245 commit fc09ad1

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

opensearch_py_ml/ml_models/crossencodermodel.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -203,7 +203,7 @@ def make_model_config_json(
203203
self,
204204
config_fname: str = "config.json",
205205
model_name: str = None,
206-
version_number: str = 1,
206+
version_number: str = '1.0.0',
207207
description: str = None,
208208
all_config: str = None,
209209
model_type: str = None,
@@ -269,10 +269,10 @@ def make_model_config_json(
269269
)
270270
model_config_content = {
271271
"name": model_name,
272-
"version": f"1.0.{version_number}",
272+
"version": version_number,
273273
"description": description,
274274
"model_format": model_format,
275-
"function_name": "TEXT_SIMILARITY",
275+
"model_mask_type": "TEXT_SIMILARITY",
276276
"model_content_hash_value": hash_value,
277277
"model_config": {
278278
"model_type": model_type,

0 commit comments

Comments
 (0)