Fixed unit tests.

lu-ohai · lu-ohai · commit 4168e878faf9 · 2025-09-19T17:35:00.000-04:00
diff --git a/tests/unitary/default_setup/model_deployment/test_model_deployment_v2.py b/tests/unitary/default_setup/model_deployment/test_model_deployment_v2.py
@@ -922,7 +922,7 @@ def test_update(
         model_deployment = self.initialize_model_deployment()
         model_deployment.dsc_model_deployment.id = "test_model_deployment_id"
         update_model_deployment_details = (
-            model_deployment._update_model_deployment_details()
+            model_deployment._update_model_deployment_details(update_type="ZDT")
         )
         model_deployment.update(wait_for_completion=True)
         mock_create.assert_called()

Original file line number	Diff line number	Diff line change
`@@ -922,7 +922,7 @@ def test_update(`
`922`	`922`	`model_deployment = self.initialize_model_deployment()`
`923`	`923`	`model_deployment.dsc_model_deployment.id = "test_model_deployment_id"`
`924`	`924`	`update_model_deployment_details = (`
`925`		`- model_deployment._update_model_deployment_details()`
	`925`	`+ model_deployment._update_model_deployment_details(update_type="ZDT")`
`926`	`926`	`)`
`927`	`927`	`model_deployment.update(wait_for_completion=True)`
`928`	`928`	`mock_create.assert_called()`