fix: unit tests

evakravi · evakravi · commit 85d372d9be91 · 2022-05-17T19:18:00.000Z
diff --git a/tests/unit/test_tuner.py b/tests/unit/test_tuner.py
@@ -1642,7 +1642,7 @@ def test_tags_prefixes_jumpstart_models(
         role=ROLE,
     )
 
-    assert sagemaker_session.create_model.call_args_list[0][0][0].startswith(
+    assert sagemaker_session.create_model.call_args_list[0][1]["name"].startswith(
         JUMPSTART_RESOURCE_BASE_NAME
     )
 
@@ -1763,7 +1763,7 @@ def test_no_tags_prefixes_non_jumpstart_models(
         role=ROLE,
     )
 
-    assert not sagemaker_session.create_model.call_args_list[0][0][0].startswith(
+    assert not sagemaker_session.create_model.call_args_list[0][1]["name"].startswith(
         JUMPSTART_RESOURCE_BASE_NAME
     )
 

Original file line number	Diff line number	Diff line change
`@@ -1642,7 +1642,7 @@ def test_tags_prefixes_jumpstart_models(`
`1642`	`1642`	`role=ROLE,`
`1643`	`1643`	`)`
`1644`	`1644`
`1645`		`- assert sagemaker_session.create_model.call_args_list[0][0][0].startswith(`
	`1645`	`+ assert sagemaker_session.create_model.call_args_list[0][1]["name"].startswith(`
`1646`	`1646`	`JUMPSTART_RESOURCE_BASE_NAME`
`1647`	`1647`	`)`
`1648`	`1648`
`@@ -1763,7 +1763,7 @@ def test_no_tags_prefixes_non_jumpstart_models(`
`1763`	`1763`	`role=ROLE,`
`1764`	`1764`	`)`
`1765`	`1765`
`1766`		`- assert not sagemaker_session.create_model.call_args_list[0][0][0].startswith(`
	`1766`	`+ assert not sagemaker_session.create_model.call_args_list[0][1]["name"].startswith(`
`1767`	`1767`	`JUMPSTART_RESOURCE_BASE_NAME`
`1768`	`1768`	`)`
`1769`	`1769`