ah the linter

Author: brandon

src/groundlight/experimental_api.py

@@ -953,9 +953,10 @@ def create_multiclass_detector(  # noqa: PLR0913 # pylint: disable=too-many-argu
         obj = self.detectors_api.create_detector(detector_creation_input, _request_timeout=DEFAULT_REQUEST_TIMEOUT)
         return Detector.parse_obj(obj.to_dict())
 
-    def _download_mlbinary_url(self, detector: Union[str, Detector]) -> str:
+    def _download_mlbinary_url(self, detector: Union[str, Detector]) -> EdgeModelInfo:
         """
-        Gets a temporary presigned URL to download the model binaries for the given detector, along with relevant metadata
+        Gets a temporary presigned URL to download the model binaries for the given detector, along
+        with relevant metadata
         """
         if isinstance(detector, Detector):
             detector = detector.id
@@ -983,14 +984,20 @@ def download_mlbinary(self, detector: Union[str, Detector], output_dir: str) ->
         def _download_and_save(url: str, output_path: str) -> None:
             try:
                 response = requests.get(url, timeout=10)
-            except:
-                raise GroundlightClientError(f"Failed to retrieve data from {url}.")
+            except Exception as e:
+                raise GroundlightClientError(f"Failed to retrieve data from {url}.") from e
             with open(output_path, "wb") as file:
                 file.write(response.content)
             return response.content
 
         if isinstance(detector, Detector):
             detector = detector.id
         edge_model_info = self._download_mlbinary_url(detector)
-        _download_and_save(edge_model_info.model_binary_url, Path(output_dir) / edge_model_info.model_binary_id)
-        _download_and_save(edge_model_info.oodd_model_binary_url, Path(output_dir) / edge_model_info.oodd_model_binary_id)
+        _download_and_save(
+            edge_model_info.model_binary_url,
+            Path(output_dir) / edge_model_info.model_binary_id
+        )
+        _download_and_save(
+            edge_model_info.oodd_model_binary_url,
+            Path(output_dir) / edge_model_info.oodd_model_binary_id
+        )