function_name -> model_mask_type

Signed-off-by: HenryL27 <[email protected]>
opensearch-project · Feb 22, 2024 · 588af41 · 588af41
1 parent 6f08d00
commit 588af41
Showing 1 changed file with 3 additions and 3 deletions.
diff --git 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,