linting

Author: bstuder

client/lomas_client/client.py

@@ -42,7 +42,8 @@
 enable_features("contrib")
 
 
-def parse_if_OK(res: requests.Response) -> IOResultE[str]:
+def parse_if_ok(res: requests.Response) -> IOResultE[str]:
+    """Only continues if Response is OK (200)."""
     if res.status_code == status.HTTP_200_OK:
         return IOSuccess(res.content.decode("utf8"))
     return IOFailure(ValueError(f"Unexpected response code: {res.status_code}: {res.content}"))
@@ -97,7 +98,7 @@ def get_dataset_metadata(self) -> IOResultE[LomasRequestModel]:
             # post to the validated body to the corresponding endpoint
             lambda body: self.http_client.post("get_dataset_metadata", body),
             # parse reply if HTTP 200
-            bind(parse_if_OK),
+            bind(parse_if_ok),
             # load successful response as json
             map_(json.loads),
         )
@@ -134,11 +135,13 @@ def get_dummy_dataset(
             # post to the validated body to the corresponding endpoint
             lambda body: self.http_client.post("get_dummy_dataset", body),
             # parse reply if HTTP 200
-            bind(parse_if_OK),
+            bind(parse_if_ok),
             # load successful response as json
             map_(DummyDsResponse.model_validate_json),
             map_(
-                lambda dummyDsRes: pl.from_pandas(dummyDsRes.dummy_df).lazy() if lazy else dummyDsRes.dummy_df
+                lambda dummy_ds_res: (
+                    pl.from_pandas(dummy_ds_res.dummy_df).lazy() if lazy else dummy_ds_res.dummy_df
+                )
             ),
         )
 
@@ -178,7 +181,7 @@ def get_initial_budget(self) -> IOResultE[InitialBudgetResponse]:
             # post to the validated body to the corresponding endpoint
             lambda body: self.http_client.post("get_initial_budget", body),
             # parse reply if HTTP 200
-            bind(parse_if_OK),
+            bind(parse_if_ok),
             # build Budget Response from successful json payload
             map_(InitialBudgetResponse.model_validate_json),
         )
@@ -198,7 +201,7 @@ def get_total_spent_budget(self) -> IOResultE[SpentBudgetResponse]:
             # post to the validated body to the corresponding endpoint
             lambda body: self.http_client.post("get_total_spent_budget", body),
             # parse reply if HTTP 200
-            bind(parse_if_OK),
+            bind(parse_if_ok),
             # build Budget Response from successful json payload
             map_(SpentBudgetResponse.model_validate_json),
         )
@@ -218,7 +221,7 @@ def get_remaining_budget(self) -> IOResultE[RemainingBudgetResponse]:
             # post to the validated body to the corresponding endpoint
             lambda body: self.http_client.post("get_remaining_budget", body),
             # parse reply if HTTP 200
-            bind(parse_if_OK),
+            bind(parse_if_ok),
             # build Budget Response from successful json payload
             map_(RemainingBudgetResponse.model_validate_json),
         )
@@ -269,7 +272,7 @@ def post_processes_queries(queries: list[dict]) -> list[dict]:
             # post to the validated body to the corresponding endpoint
             lambda body: self.http_client.post("get_previous_queries", body),
             # parse reply if HTTP 200
-            bind(parse_if_OK),
+            bind(parse_if_ok),
             bind(lambda content: json.loads(content)["previous_queries"]),
             post_processes_queries,
         )