fix: list_models() for python3.8 (#4756)

malav-shastri · Malav Shastri · web-flow · commit 31190c4638ae · 2024-06-24T21:01:21.000-07:00
* fix: list_models() for python3.8

* fix linting

---------

Co-authored-by: Malav Shastri &lt;malavhs@amazon.com&gt;
diff --git a/src/sagemaker/jumpstart/hub/hub.py b/src/sagemaker/jumpstart/hub/hub.py
@@ -184,13 +184,16 @@ def list_models(self, clear_cache: bool = True, **kwargs) -> Dict[str, Any]:
                 **{
                     "hub_name": self.hub_name,
                     "hub_content_type": HubContentType.MODEL_REFERENCE.value,
+                    **kwargs,
                 }
-                | kwargs
             )
 
             hub_model_summaries = self._list_and_paginate_models(
-                **{"hub_name": self.hub_name, "hub_content_type": HubContentType.MODEL.value}
-                | kwargs
+                **{
+                    "hub_name": self.hub_name,
+                    "hub_content_type": HubContentType.MODEL.value,
+                    **kwargs,
+                }
             )
             response["hub_content_summaries"] = hub_model_reference_summaries + hub_model_summaries
         response["next_token"] = None  # Temporary until pagination is implemented

Original file line number	Diff line number	Diff line change
`@@ -184,13 +184,16 @@ def list_models(self, clear_cache: bool = True, **kwargs) -> Dict[str, Any]:`
`184`	`184`	`**{`
`185`	`185`	`"hub_name": self.hub_name,`
`186`	`186`	`"hub_content_type": HubContentType.MODEL_REFERENCE.value,`
	`187`	`+ **kwargs,`
`187`	`188`	`}`
`188`		`- \| kwargs`
`189`	`189`	`)`
`190`	`190`
`191`	`191`	`hub_model_summaries = self._list_and_paginate_models(`
`192`		`- **{"hub_name": self.hub_name, "hub_content_type": HubContentType.MODEL.value}`
`193`		`- \| kwargs`
	`192`	`+ **{`
	`193`	`+ "hub_name": self.hub_name,`
	`194`	`+ "hub_content_type": HubContentType.MODEL.value,`
	`195`	`+ **kwargs,`
	`196`	`+ }`
`194`	`197`	`)`
`195`	`198`	`response["hub_content_summaries"] = hub_model_reference_summaries + hub_model_summaries`
`196`	`199`	`response["next_token"] = None # Temporary until pagination is implemented`