lint: ruff

Author: shamrin

datacrunch/InferenceClient/inference_client.py

@@ -81,7 +81,7 @@ def output(self, is_text: bool = False) -> Any:
                 raise InferenceClientError(
                     'Response might be a stream, use the stream method instead'
                 ) from e
-            raise InferenceClientError(f'Failed to parse response as JSON: {str(e)}') from e
+            raise InferenceClientError(f'Failed to parse response as JSON: {e!s}') from e
 
     def stream(self, chunk_size: int = 512, as_text: bool = True) -> Generator[Any, None, None]:
         """Stream the response content.
@@ -228,7 +228,7 @@ def _make_request(self, method: str, path: str, **kwargs) -> requests.Response:
                 f'Request to {path} timed out after {timeout} seconds'
             ) from e
         except requests.exceptions.RequestException as e:
-            raise InferenceClientError(f'Request to {path} failed: {str(e)}') from e
+            raise InferenceClientError(f'Request to {path} failed: {e!s}') from e
 
     def run_sync(
         self,
@@ -458,7 +458,7 @@ def health(self, healthcheck_path: str = '/health') -> requests.Response:
         try:
             return self.get(healthcheck_path)
         except InferenceClientError as e:
-            raise InferenceClientError(f'Health check failed: {str(e)}') from e
+            raise InferenceClientError(f'Health check failed: {e!s}') from e
 
 
 @dataclass_json(undefined=Undefined.EXCLUDE)

datacrunch/containers/containers.py

@@ -386,7 +386,7 @@ def __str__(self):
         # Get all attributes except _inference_client
         attrs = {k: v for k, v in self.__dict__.items() if k != '_inference_client'}
         # Format each attribute
-        attr_strs = [f'{k}={repr(v)}' for k, v in attrs.items()]
+        attr_strs = [f'{k}={v!r}' for k, v in attrs.items()]
         return f'Deployment({", ".join(attr_strs)})'
 
     def __repr__(self):
@@ -399,7 +399,7 @@ def __repr__(self):
 
     @classmethod
     def from_dict_with_inference_key(
-        cls, data: dict[str, Any], inference_key: str = None
+        cls, data: dict[str, Any], inference_key: str | None = None
     ) -> 'Deployment':
         """Creates a Deployment instance from a dictionary with an inference key.
 
@@ -731,7 +731,7 @@ class ContainersService:
     deployment API, including CRUD operations for deployments and related resources.
     """
 
-    def __init__(self, http_client: HTTPClient, inference_key: str = None) -> None:
+    def __init__(self, http_client: HTTPClient, inference_key: str | None = None) -> None:
         """Initializes the containers service.
 
         Args:
@@ -987,7 +987,7 @@ def delete_deployment_environment_variables(
         return result
 
     def get_compute_resources(
-        self, size: int = None, is_available: bool = None
+        self, size: int | None = None, is_available: bool | None = None
     ) -> list[ComputeResource]:
         """Retrieves compute resources, optionally filtered by size and availability.
 

datacrunch/datacrunch.py

@@ -22,7 +22,7 @@ def __init__(
         client_id: str,
         client_secret: str,
         base_url: str = 'https://api.datacrunch.io/v1',
-        inference_key: str = None,
+        inference_key: str | None = None,
     ) -> None:
         """The DataCrunch client.
 

datacrunch/helpers.py

@@ -10,8 +10,8 @@ def stringify_class_object_properties(class_object: type) -> str:
     :rtype: json string representation of a class object's properties and values
     """
     class_properties = {
-        property: getattr(class_object, property, '') # noqa: A001
-        for property in class_object.__dir__() # noqa: A001
+        property: getattr(class_object, property, '')
+        for property in class_object.__dir__()  # noqa: A001
         if property[:1] != '_' and type(getattr(class_object, property, '')).__name__ != 'method'
     }
     return json.dumps(class_properties, indent=2)

datacrunch/http_client/http_client.py

@@ -33,7 +33,9 @@ def __init__(self, auth_service, base_url: str) -> None:
         self._auth_service = auth_service
         self._auth_service.authenticate()
 
-    def post(self, url: str, json: dict = None, params: dict = None, **kwargs) -> requests.Response:
+    def post(
+        self, url: str, json: dict | None = None, params: dict | None = None, **kwargs
+    ) -> requests.Response:
         """Sends a POST request.
 
         A wrapper for the requests.post method.
@@ -62,7 +64,9 @@ def post(self, url: str, json: dict = None, params: dict = None, **kwargs) -> re
 
         return response
 
-    def put(self, url: str, json: dict = None, params: dict = None, **kwargs) -> requests.Response:
+    def put(
+        self, url: str, json: dict | None = None, params: dict | None = None, **kwargs
+    ) -> requests.Response:
         """Sends a PUT request.
 
         A wrapper for the requests.put method.
@@ -91,7 +95,7 @@ def put(self, url: str, json: dict = None, params: dict = None, **kwargs) -> req
 
         return response
 
-    def get(self, url: str, params: dict = None, **kwargs) -> requests.Response:
+    def get(self, url: str, params: dict | None = None, **kwargs) -> requests.Response:
         """Sends a GET request.
 
         A wrapper for the requests.get method.
@@ -119,7 +123,7 @@ def get(self, url: str, params: dict = None, **kwargs) -> requests.Response:
         return response
 
     def patch(
-        self, url: str, json: dict = None, params: dict = None, **kwargs
+        self, url: str, json: dict | None = None, params: dict | None = None, **kwargs
     ) -> requests.Response:
         """Sends a PATCH request.
 
@@ -150,7 +154,7 @@ def patch(
         return response
 
     def delete(
-        self, url: str, json: dict = None, params: dict = None, **kwargs
+        self, url: str, json: dict | None = None, params: dict | None = None, **kwargs
     ) -> requests.Response:
         """Sends a DELETE request.
 

datacrunch/volumes/volumes.py

@@ -16,11 +16,11 @@ def __init__(
         type: str,
         is_os_volume: bool,
         created_at: str,
-        target: str = None,
+        target: str | None = None,
         location: str = Locations.FIN_03,
-        instance_id: str = None,
+        instance_id: str | None = None,
         ssh_key_ids: list[str] = [],
-        deleted_at: str = None,
+        deleted_at: str | None = None,
     ) -> None:
         """Initialize the volume object.
 
@@ -209,7 +209,7 @@ class VolumesService:
     def __init__(self, http_client) -> None:
         self._http_client = http_client
 
-    def get(self, status: str = None) -> list[Volume]:
+    def get(self, status: str | None = None) -> list[Volume]:
         """Get all of the client's non-deleted volumes, or volumes with specific status.
 
         :param status: optional, status of the volumes, defaults to None
@@ -247,7 +247,7 @@ def create(
         type: str,
         name: str,
         size: int,
-        instance_id: str = None,
+        instance_id: str | None = None,
         location: str = Locations.FIN_03,
     ) -> Volume:
         """Create new volume.
@@ -311,7 +311,7 @@ def detach(self, id_list: list[str] | str) -> None:
         self._http_client.put(VOLUMES_ENDPOINT, json=payload)
         return
 
-    def clone(self, id: str, name: str = None, type: str = None) -> Volume:
+    def clone(self, id: str, name: str | None = None, type: str | None = None) -> Volume:
         """Clone a volume or multiple volumes.
 
         :param id: volume id or list of volume ids