format all

Author: shamrin

datacrunch/InferenceClient/inference_client.py

@@ -1,5 +1,6 @@
 try:
     from importlib.metadata import version
+
     __version__ = version('datacrunch')
 except Exception:
-    __version__ = "0.0.0+dev" # fallback for development
+    __version__ = '0.0.0+dev'  # fallback for development

datacrunch/_version.py

@@ -34,13 +34,12 @@ def authenticate(self) -> dict:
         """
         url = self._base_url + TOKEN_ENDPOINT
         payload = {
-            "grant_type": CLIENT_CREDENTIALS,
-            "client_id": self._client_id,
-            "client_secret": self._client_secret
+            'grant_type': CLIENT_CREDENTIALS,
+            'client_id': self._client_id,
+            'client_secret': self._client_secret,
         }
 
-        response = requests.post(
-            url, json=payload, headers=self._generate_headers())
+        response = requests.post(url, json=payload, headers=self._generate_headers())
         handle_error(response)
 
         auth_data = response.json()
@@ -71,13 +70,9 @@ def refresh(self) -> dict:
         """
         url = self._base_url + TOKEN_ENDPOINT
 
-        payload = {
-            "grant_type": REFRESH_TOKEN,
-            "refresh_token": self._refresh_token
-        }
+        payload = {'grant_type': REFRESH_TOKEN, 'refresh_token': self._refresh_token}
 
-        response = requests.post(
-            url, json=payload, headers=self._generate_headers())
+        response = requests.post(url, json=payload, headers=self._generate_headers())
 
         # if refresh token is also expired, authenticate again:
         if response.status_code == 401 or response.status_code == 400:
@@ -98,9 +93,7 @@ def refresh(self) -> dict:
     def _generate_headers(self):
         # get the first 10 chars of the client id
         client_id_truncated = self._client_id[:10]
-        headers = {
-            'User-Agent': 'datacrunch-python-' + client_id_truncated
-        }
+        headers = {'User-Agent': 'datacrunch-python-' + client_id_truncated}
         return headers
 
     def is_expired(self) -> bool:

datacrunch/authentication/authentication.py

@@ -1,4 +1,3 @@
-
 BALANCE_ENDPOINT = '/balance'
 
 
@@ -48,4 +47,4 @@ def get(self) -> Balance:
         :rtype: Balance
         """
         balance = self._http_client.get(BALANCE_ENDPOINT).json()
-        return Balance(balance["amount"], balance["currency"])
+        return Balance(balance['amount'], balance['currency'])

datacrunch/balance/balance.py

@@ -36,42 +36,42 @@ def __init__(self):
 
 
 class VolumeStatus:
-    ORDERED = "ordered"
-    CREATING = "creating"
-    ATTACHED = "attached"
-    DETACHED = "detached"
-    DELETING = "deleting"
-    DELETED = "deleted"
+    ORDERED = 'ordered'
+    CREATING = 'creating'
+    ATTACHED = 'attached'
+    DETACHED = 'detached'
+    DELETING = 'deleting'
+    DELETED = 'deleted'
     CLONING = 'cloning'
 
     def __init__(self):
         return
 
 
 class VolumeTypes:
-    NVMe = "NVMe"
-    HDD = "HDD"
+    NVMe = 'NVMe'
+    HDD = 'HDD'
 
     def __init__(self):
         return
 
 
 class Locations:
-    FIN_01: str = "FIN-01"
-    ICE_01: str = "ICE-01"
+    FIN_01: str = 'FIN-01'
+    ICE_01: str = 'ICE-01'
 
     def __init__(self):
         return
 
 
 class ErrorCodes:
-    INVALID_REQUEST = "invalid_request"
-    UNAUTHORIZED_REQUEST = "unauthorized_request"
-    INSUFFICIENT_FUNDS = "insufficient_funds"
-    FORBIDDEN_ACTION = "forbidden_action"
-    NOT_FOUND = "not_found"
-    SERVER_ERROR = "server_error"
-    SERVICE_UNAVAILABLE = "service_unavailable"
+    INVALID_REQUEST = 'invalid_request'
+    UNAUTHORIZED_REQUEST = 'unauthorized_request'
+    INSUFFICIENT_FUNDS = 'insufficient_funds'
+    FORBIDDEN_ACTION = 'forbidden_action'
+    NOT_FOUND = 'not_found'
+    SERVER_ERROR = 'server_error'
+    SERVICE_UNAVAILABLE = 'service_unavailable'
 
     def __init__(self):
         return