fixed unit tests

Author: Aryanag2

test_mmd.py

@@ -0,0 +1,31 @@
+import ads
+from ads.model.datascience_model_group import DataScienceModelGroup
+import json
+
+# 1. Set Auth with the specific profile
+ads.set_auth(auth="security_token", profile="aryan-ashburn2")
+
+# 2. Get the Model Group created by your deployment (from latest logs)
+group_id = "ocid1.datasciencemodelgroupint.oc1.iad.amaaaaaav66vvniafod77ugys4lya3xsq75frfpxzjbjbcipohli6pibik3q"
+
+try:
+    model_group = DataScienceModelGroup.from_id(group_id)
+
+    # 3. Extract the configuration
+    config_value = model_group.custom_metadata_list.get("MULTI_MODEL_CONFIG").value
+    config_json = json.loads(config_value)
+
+    # 4. Print and Verify
+    print("\n--- Verification Results ---")
+    for model in config_json['models']:
+        print(f"\nModel Name: {model.get('model_name', 'Unknown')}")
+        print(f"Params:     {model['params']}")
+        
+        if "--max-model-len" in model['params']:
+            print(">> STATUS: Has SMM Defaults (Expected for 'Llama_Default2')")
+        else:
+            print(">> STATUS: Clean / No Defaults (Expected for 'Llama_Clear2')")
+
+except Exception as e:
+    print(f"Error fetching model group: {e}")
+

test_mmd2.py

@@ -0,0 +1,32 @@
+import ads
+from ads.model.datascience_model_group import DataScienceModelGroup
+import json
+
+# 1. Set Auth with the specific profile
+ads.set_auth(auth="security_token", profile="aryan-ashburn2")
+
+# 2. Get the Model Group created by your deployment (from latest logs)
+group_id = "ocid1.datasciencemodelgroupint.oc1.iad.amaaaaaav66vvniazm2a2ao2u7n65baecxtu6e6lejfvj7gb3ytu3zduq35q"
+
+try:
+    model_group = DataScienceModelGroup.from_id(group_id)
+
+    # 3. Extract the configuration
+    config_value = model_group.custom_metadata_list.get("MULTI_MODEL_CONFIG").value
+    config_json = json.loads(config_value)
+
+    # 4. Print and Verify
+    print("\n--- Verification Results ---")
+    for model in config_json['models']:
+        print(f"\nModel Name: {model.get('model_name', 'Unknown')}")
+        print(f"Params:     {model['params']}")
+        
+        if "--max-model-len 1024" in model['params']:
+            print(">> STATUS: SUCCESS - Custom value used (1024)")
+        elif "--max-model-len 65536" in model['params']:
+            print(">> STATUS: FAIL - Defaults merged in (65536)")
+        else:
+            print(">> STATUS: FAIL - Param missing entirely")
+
+except Exception as e:
+    print(f"Error fetching model group: {e}")

tests/unitary/with_extras/aqua/test_common_entities.py

@@ -196,7 +196,7 @@ def test_extract_params_from_env_var_missing_env(self):
         }
         result = AquaMultiModelRef.model_validate(values)
         assert result.env_var == {}
-        assert result.params == {}
+        assert result.params is None
 
     def test_all_model_ids_no_finetunes(self):
         model = AquaMultiModelRef(model_id="ocid1.model.oc1..base")

tests/unitary/with_extras/aqua/test_deployment.py

@@ -556,7 +556,7 @@ class TestDataset:
         "models": [
             {
                 "env_var": {},
-                "params": {},
+                "params": None,
                 "gpu_count": 2,
                 "model_id": "test_model_id_1",
                 "model_name": "test_model_1",
@@ -566,7 +566,7 @@ class TestDataset:
             },
             {
                 "env_var": {},
-                "params": {},
+                "params": None,
                 "gpu_count": 2,
                 "model_id": "test_model_id_2",
                 "model_name": "test_model_2",
@@ -576,7 +576,7 @@ class TestDataset:
             },
             {
                 "env_var": {},
-                "params": {},
+                "params": None,
                 "gpu_count": 2,
                 "model_id": "test_model_id_3",
                 "model_name": "test_model_3",
@@ -1258,9 +1258,7 @@ def test_get_deployment(self, mock_get_resource_name):
         mock_get_resource_name.side_effect = lambda param: (
             "log-group-name"
             if param.startswith("ocid1.loggroup")
-            else "log-name"
-            if param.startswith("ocid1.log")
-            else ""
+            else "log-name" if param.startswith("ocid1.log") else ""
         )
 
         result = self.app.get(model_deployment_id=TestDataset.MODEL_DEPLOYMENT_ID)
@@ -1301,9 +1299,7 @@ def test_get_multi_model_deployment(
         mock_get_resource_name.side_effect = lambda param: (
             "log-group-name"
             if param.startswith("ocid1.loggroup")
-            else "log-name"
-            if param.startswith("ocid1.log")
-            else ""
+            else "log-name" if param.startswith("ocid1.log") else ""
         )
 
         aqua_multi_model = os.path.join(