please the linters

enlyze · Dec 1, 2023 · 8499a86 · 8499a86
1 parent 80b1a87
commit 8499a86
Show file tree

Hide file tree

Showing 6 changed files with 7 additions and 8 deletions.
diff --git a/src/enlyze/api_clients/base.py b/src/enlyze/api_clients/base.py
@@ -9,7 +9,7 @@
 from pydantic import BaseModel, ValidationError
 
 from enlyze.auth import TokenAuth
-from enlyze.constants import ENLYZE_BASE_URL, HTTPX_TIMEOUT
+from enlyze.constants import HTTPX_TIMEOUT
 from enlyze.errors import EnlyzeError, InvalidTokenError
 
 

diff --git a/src/enlyze/api_clients/production_runs/client.py b/src/enlyze/api_clients/production_runs/client.py
@@ -4,7 +4,7 @@
 from pydantic import BaseModel
 
 from enlyze.api_clients.base import ApiBaseClient, PaginatedResponseBaseModel
-from enlyze.constants import ENLYZE_BASE_URL, PRODUCTION_RUNS_API_SUB_PATH
+from enlyze.constants import PRODUCTION_RUNS_API_SUB_PATH
 
 
 class _Metadata(BaseModel):

diff --git a/src/enlyze/api_clients/timeseries/client.py b/src/enlyze/api_clients/timeseries/client.py
@@ -4,7 +4,7 @@
 from pydantic import AnyUrl
 
 from enlyze.api_clients.base import ApiBaseClient, PaginatedResponseBaseModel
-from enlyze.constants import ENLYZE_BASE_URL, TIMESERIES_API_SUB_PATH
+from enlyze.constants import TIMESERIES_API_SUB_PATH
 
 
 class _PaginatedResponse(PaginatedResponseBaseModel):

diff --git a/src/enlyze/client.py b/src/enlyze/client.py
@@ -52,14 +52,14 @@ class EnlyzeClient:
 
     """
 
-    def __init__(self, token: str, base_url: str | None = ENLYZE_BASE_URL) -> None:
+    def __init__(self, token: str, base_url: str | None = None) -> None:
         self._timeseries_api_client = TimeseriesApiClient(
             token=token,
-            base_url=base_url,
+            base_url=base_url or ENLYZE_BASE_URL,
         )
         self._production_runs_api_client = ProductionRunsApiClient(
             token=token,
-            base_url=base_url,
+            base_url=base_url or ENLYZE_BASE_URL,
         )
 
     def _get_sites(self) -> Iterator[timeseries_api_models.Site]:

diff --git a/tests/enlyze/api_clients/production_runs/test_client.py b/tests/enlyze/api_clients/production_runs/test_client.py
@@ -7,7 +7,7 @@
     _Metadata,
     _PaginatedResponse,
 )
-from enlyze.constants import ENLYZE_BASE_URL, PRODUCTION_RUNS_API_SUB_PATH
+from enlyze.constants import PRODUCTION_RUNS_API_SUB_PATH
 
 
 @pytest.fixture

diff --git a/tests/enlyze/api_clients/test_base.py b/tests/enlyze/api_clients/test_base.py
@@ -12,7 +12,6 @@
     ApiBaseModel,
     PaginatedResponseBaseModel,
 )
-from enlyze.constants import ENLYZE_BASE_URL
 from enlyze.errors import EnlyzeError, InvalidTokenError