lint

Author: bencrabtree

src/sagemaker/jumpstart/types.py

@@ -1363,7 +1363,8 @@ def from_json(self, json_obj: Dict[str, Any]) -> None:
         self.deploy_kwargs = deepcopy(json_obj.get("deploy_kwargs", {}))
         self.predictor_specs: Optional[JumpStartPredictorSpecs] = (
             JumpStartPredictorSpecs(
-                json_obj.get("sage_maker_sdk_predictor_specifications"), is_hub_content=self._is_hub_content
+                json_obj.get("sage_maker_sdk_predictor_specifications"),
+                is_hub_content=self._is_hub_content,
             )
             if json_obj.get("sage_maker_sdk_predictor_specifications")
             else None

tests/unit/sagemaker/jumpstart/hub/test_hub.py

@@ -218,7 +218,10 @@ def test_describe_model_one_thrown_error(mock_describe_hub_content_response, sag
 
         mock_describe_hub_content.asssert_called_times(2)
         mock_describe_hub_content.assert_called_with(
-            hub_name=HUB_NAME, hub_content_name="test-model", hub_content_version="3.0", hub_content_type="Model"
+            hub_name=HUB_NAME,
+            hub_content_name="test-model",
+            hub_content_version="3.0",
+            hub_content_type="Model",
         )