From 9f3030e5c6cacb71bc5a45e4d0b8a7fd4fc7d122 Mon Sep 17 00:00:00 2001 From: Azory YData Bot Date: Thu, 14 Dec 2023 14:53:44 +0000 Subject: [PATCH] fix(linting): code formatting --- src/ydata/sdk/common/client/client.py | 6 ++++-- src/ydata/sdk/synthesizers/synthesizer.py | 15 ++++++++++----- 2 files changed, 14 insertions(+), 7 deletions(-) diff --git a/src/ydata/sdk/common/client/client.py b/src/ydata/sdk/common/client/client.py index 55b9db32..a3a3a258 100644 --- 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 index 0ea4bd4e..14ce5e75 100644 --- 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())