From 588af41a0156fbee321616f2431ece58e9968bf3 Mon Sep 17 00:00:00 2001 From: HenryL27 Date: Wed, 21 Feb 2024 14:43:37 -0800 Subject: [PATCH] function_name -> model_mask_type Signed-off-by: HenryL27 --- opensearch_py_ml/ml_models/crossencodermodel.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/opensearch_py_ml/ml_models/crossencodermodel.py b/opensearch_py_ml/ml_models/crossencodermodel.py index da8f229c..37e997fe 100644 --- a/opensearch_py_ml/ml_models/crossencodermodel.py +++ b/opensearch_py_ml/ml_models/crossencodermodel.py @@ -203,7 +203,7 @@ def make_model_config_json( self, config_fname: str = "config.json", model_name: str = None, - version_number: str = 1, + version_number: str = '1.0.0', description: str = None, all_config: str = None, model_type: str = None, @@ -269,10 +269,10 @@ def make_model_config_json( ) model_config_content = { "name": model_name, - "version": f"1.0.{version_number}", + "version": version_number, "description": description, "model_format": model_format, - "function_name": "TEXT_SIMILARITY", + "model_mask_type": "TEXT_SIMILARITY", "model_content_hash_value": hash_value, "model_config": { "model_type": model_type,