lint

Author: bencrabtree

src/sagemaker/jumpstart/types.py

@@ -1285,6 +1285,7 @@ def from_json(self, json_obj: Dict[str, Any]) -> None:
         Args:
             json_obj (Dict[str, Any]): Dictionary representation of spec.
         """
+        print("HEYHEYHEY")
         if self._is_hub_content:
             json_obj = walk_and_apply_json(json_obj, camel_to_snake)
         self.model_id: str = json_obj.get("model_id")
@@ -1361,6 +1362,7 @@ def from_json(self, json_obj: Dict[str, Any]) -> None:
             )
         self.model_kwargs = deepcopy(json_obj.get("model_kwargs", {}))
         self.deploy_kwargs = deepcopy(json_obj.get("deploy_kwargs", {}))
+        print(f'HUH {json_obj.get("predictor_specs")}')
         self.predictor_specs: Optional[JumpStartPredictorSpecs] = (
             JumpStartPredictorSpecs(
                 json_obj.get("predictor_specs"),
@@ -1535,11 +1537,11 @@ def from_json(self, json_obj: Dict[str, Any]) -> None:
         for field in json_obj.keys():
             if field in self.__slots__:
                 setattr(self, field, json_obj[field])
-        
+
         # Handle custom fields
-        for field, custom_field in self.CUSTOM_FIELD_MAP.items():
+        for custom_field, field in self.CUSTOM_FIELD_MAP.items():
             if field in json_obj:
-                setattr(self, custom_field, json_obj[field])
+                setattr(self, field, json_obj[custom_field])
 
 
 class JumpStartMetadataConfig(JumpStartDataHolderType):
@@ -1732,7 +1734,9 @@ def __init__(self, spec: Dict[str, Any], is_hub_content: Optional[bool] = False)
             spec (Dict[str, Any]): Dictionary representation of spec.
             is_hub_content (Optional[bool]): Whether the model is from a private hub.
         """
+        print("here?")
         super().__init__(spec, is_hub_content)
+        print("hi")
         self.from_json(spec)
         if self.inference_configs and self.inference_configs.get_top_config_from_ranking():
             super().from_json(self.inference_configs.get_top_config_from_ranking().resolved_config)

tests/unit/sagemaker/jumpstart/utils.py

@@ -144,6 +144,7 @@ def get_special_model_spec(
     """
 
     specs = JumpStartModelSpecs(SPECIAL_MODEL_SPECS_DICT[model_id])
+    print(f"HERE {specs.predictor_specs}")
     return specs