fix failing unit tests

Malav Shastri · Malav Shastri · commit 38495dc1ecfd · 2024-07-09T02:30:38.000-04:00
diff --git a/tests/unit/sagemaker/jumpstart/model/test_model.py b/tests/unit/sagemaker/jumpstart/model/test_model.py
@@ -1360,7 +1360,6 @@ def test_attach(
             model_id="model-id",
             model_version="model-version",
             sagemaker_session=DEFAULT_JUMPSTART_SAGEMAKER_SESSION,
-            hub_name=None,
         )
         assert isinstance(val, JumpStartModel)
 

Original file line number	Diff line number	Diff line change
`@@ -1360,7 +1360,6 @@ def test_attach(`
`1360`	`1360`	`model_id="model-id",`
`1361`	`1361`	`model_version="model-version",`
`1362`	`1362`	`sagemaker_session=DEFAULT_JUMPSTART_SAGEMAKER_SESSION,`
`1363`		`- hub_name=None,`
`1364`	`1363`	`)`
`1365`	`1364`	`assert isinstance(val, JumpStartModel)`
`1366`	`1365`