fix(linting): code formatting

ydataai · Dec 14, 2023 · 9f3030e · 9f3030e
1 parent f7e9ec7
commit 9f3030e
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 7 deletions.
diff --git a/src/ydata/sdk/common/client/client.py b/src/ydata/sdk/common/client/client.py
@@ -74,7 +74,8 @@ def post(self, endpoint: str, data: Optional[Dict] = None, json: Optional[Dict]
         Returns:
             Response object
         """
-        url_data = self.__build_url(endpoint, data=data, json=json, files=files, project=project)
+        url_data = self.__build_url(
+            endpoint, data=data, json=json, files=files, project=project)
         response = self._http_client.post(**url_data)
 
         if response.status_code != Client.codes.OK and raise_for_status:
@@ -94,7 +95,8 @@ def get(self, endpoint: str, params: Optional[Dict] = None,
         Returns:
             Response object
         """
-        url_data = self.__build_url(endpoint, params=params, cookies=cookies, project=project)
+        url_data = self.__build_url(endpoint, params=params,
+                                    cookies=cookies, project=project)
         response = self._http_client.get(**url_data)
 
         if response.status_code != Client.codes.OK and raise_for_status:

diff --git a/src/ydata/sdk/synthesizers/synthesizer.py b/src/ydata/sdk/synthesizers/synthesizer.py
@@ -53,7 +53,8 @@ class BaseSynthesizer(ABC, ModelFactoryMixin):
 
     def __init__(self, uid: UID | None = None, name: str | None = None, project: Project | None = None, client: Client | None = None):
         self._init_common(client=client)
-        self._model = mSynthesizer(uid=uid, name=name or str(uuid4())) if uid or project else None
+        self._model = mSynthesizer(uid=uid, name=name or str(
+            uuid4())) if uid or project else None
         self.__project = project
 
     @init_client
@@ -240,7 +241,8 @@ def _fit_from_datasource(
         if condition_on is not None:
             payload["extraData"]["condition_on"] = condition_on
 
-        response = self._client.post('/synthesizer/', json=payload, project=self.__project)
+        response = self._client.post(
+            '/synthesizer/', json=payload, project=self.__project)
         data: list = response.json()
         self._model, _ = self._model_from_api(X.datatype, data)
         while self.status not in [Status.READY, Status.FAILED]:
@@ -271,14 +273,16 @@ def _sample(self, payload: Dict) -> pdDataFrame:
         Returns:
             pandas `DataFrame`
         """
-        response = self._client.post(f"/synthesizer/{self.uid}/sample", json=payload, project=self.__project)
+        response = self._client.post(
+            f"/synthesizer/{self.uid}/sample", json=payload, project=self.__project)
 
         data: Dict = response.json()
         sample_uid = data.get('uid')
         sample_status = None
         while sample_status not in ['finished', 'failed']:
             self._logger.info('Sampling from the synthesizer...')
-            response = self._client.get(f'/synthesizer/{self.uid}/history', project=self.__project)
+            response = self._client.get(
+                f'/synthesizer/{self.uid}/history', project=self.__project)
             history: Dict = response.json()
             sample_data = next((s for s in history if s.get('uid') == sample_uid), None)
             sample_status = sample_data.get('status', {}).get('state')
@@ -318,7 +322,8 @@ def status(self) -> Status:
             return Status.UNKNOWN
 
     def get(self):
-        assert self._is_initialized() and self._model.uid, InputError("Please provide the synthesizer `uid`")
+        assert self._is_initialized() and self._model.uid, InputError(
+            "Please provide the synthesizer `uid`")
 
         response = self._client.get(f'/synthesizer/{self.uid}', project=self.__project)
         data = filter_dict(mSynthesizer, response.json())