formatting the baseModelId (#1601)

Azure · Oct 31, 2023 · c698075 · c698075
1 parent 9283a1d
commit c698075
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/assets/training/finetune_acft_hf_nlp/src/register_model/register_model.py b/assets/training/finetune_acft_hf_nlp/src/register_model/register_model.py
@@ -170,11 +170,12 @@ def get_properties(finetune_args_path: str) -> Dict[str, str]:
     }
     for property_key, finetune_args_key in property_key_to_finetune_args_key_map.items():
         properties[property_key] = finetune_args_dict.get(finetune_args_key, None)
+        if "baseModelId" == property_key:
+            properties[property_key] = "/".join(properties[property_key].split('/')[:-2])
 
     # fixed properties
     additional_properties = {
         "baseModelWeightsVersion": 1.0,
-        "azureml.datastoreId": "datastore_id"
     }
     properties.update(additional_properties)
     logger.info(f"Adding the following properties to the registered model: {properties}")