style: format with black and remove unuse var

Author: corentin-hrflow

hrflow/hrflow/text/tagging.py

@@ -70,7 +70,7 @@ def post(
             output_lang=output_lang,
             top_n=top_n,
         )
-        
+
         if texts is None and text is not None:
             payload["text"] = text
         elif text is None and texts is not None:
@@ -79,6 +79,6 @@ def post(
             raise ValueError("Either text or texts must be provided.")
         else:
             raise ValueError("Only one of text or texts must be provided.")
-        
+
         response = self.client.post("text/tagging", json=payload)
         return validate_response(response)

tests/test_job.py

@@ -20,13 +20,15 @@
     _var_from_env_get,
 )
 
+
 @pytest.fixture(scope="module")
 def hrflow_client():
     return Hrflow(
         api_secret=_var_from_env_get("HRFLOW_API_KEY"),
         api_user=_var_from_env_get("HRFLOW_USER_EMAIL"),
     )
 
+
 def _job_get() -> t.Dict[str, t.Any]:
     return dict(
         reference=str(uuid4()),

tests/test_profile.py

@@ -29,13 +29,15 @@
 _ASYNC_RETRY_INTERVAL_SECONDS = 5
 _ASYNC_TIMEOUT_SECONDS = 60
 
+
 @pytest.fixture(scope="module")
 def hrflow_client():
     return Hrflow(
         api_secret=_var_from_env_get("HRFLOW_API_KEY"),
         api_user=_var_from_env_get("HRFLOW_USER_EMAIL"),
     )
 
+
 def _profile_get() -> t.Dict[str, t.Any]:
     return dict(
         reference=str(uuid4()),
@@ -330,7 +332,9 @@ def test_profile_parsing_file_quicksilver_async_basic(hrflow_client):
     assert _ASYNC_RETRY_INTERVAL_SECONDS > 0
     for _ in range(max(0, _ASYNC_TIMEOUT_SECONDS // _ASYNC_RETRY_INTERVAL_SECONDS)):
         model = ProfileIndexingResponse.model_validate(
-            hrflow_client.profile.storing.get(source_key=SOURCE_KEY, reference=reference)
+            hrflow_client.profile.storing.get(
+                source_key=SOURCE_KEY, reference=reference
+            )
         )
         if model.code == http_codes.ok:
             break
@@ -408,7 +412,9 @@ def test_profile_parsing_file_mozart_async_basic(hrflow_client):
     assert _ASYNC_RETRY_INTERVAL_SECONDS > 0
     for _ in range(max(0, _ASYNC_TIMEOUT_SECONDS // _ASYNC_RETRY_INTERVAL_SECONDS)):
         model = ProfileIndexingResponse.model_validate(
-            hrflow_client.profile.storing.get(source_key=SOURCE_KEY, reference=reference)
+            hrflow_client.profile.storing.get(
+                source_key=SOURCE_KEY, reference=reference
+            )
         )
         if model.code == http_codes.ok:
             break
@@ -514,7 +520,9 @@ def test_profile_asking_basic(hrflow_client):
     model = ProfileAskingResponse.model_validate(
         hrflow_client.profile.asking.get(
             source_key=SOURCE_KEY,
-            key=_indexed_response_get(hrflow_client, SOURCE_KEY, _profile_get()).data.key,
+            key=_indexed_response_get(
+                hrflow_client, SOURCE_KEY, _profile_get()
+            ).data.key,
             questions=[
                 "What is the full name of the profile ?",
             ],
@@ -539,7 +547,9 @@ def test_profile_asking_multiple_questions(hrflow_client):
         hrflow_client.profile.asking.get(
             source_key=SOURCE_KEY,
             questions=questions,
-            key=_indexed_response_get(hrflow_client, SOURCE_KEY, _profile_get()).data.key,
+            key=_indexed_response_get(
+                hrflow_client, SOURCE_KEY, _profile_get()
+            ).data.key,
         )
     )
     assert model.code == http_codes.ok
@@ -556,7 +566,9 @@ def test_profile_asking_no_question(hrflow_client):
     model = ProfileAskingResponse.model_validate(
         hrflow_client.profile.asking.get(
             source_key=SOURCE_KEY,
-            key=_indexed_response_get(hrflow_client, SOURCE_KEY, _profile_get()).data.key,
+            key=_indexed_response_get(
+                hrflow_client, SOURCE_KEY, _profile_get()
+            ).data.key,
             questions=None,
         )
     )

tests/test_text.py

@@ -33,13 +33,15 @@
     ),
 ]
 
+
 @pytest.fixture(scope="module")
 def hrflow_client():
     return Hrflow(
         api_secret=_var_from_env_get("HRFLOW_API_KEY"),
         api_user=_var_from_env_get("HRFLOW_USER_EMAIL"),
     )
 
+
 @pytest.mark.text
 @pytest.mark.embedding
 def test_embedding_basic(hrflow_client):
@@ -171,6 +173,7 @@ def test_tagger_rome_family_with_text_param(hrflow_client):
     assert model.code == requests.codes.ok
     assert isinstance(model.data, TextTaggingDataItem)
 
+
 @pytest.mark.text
 @pytest.mark.tagging
 def test_tagger_rome_family_with_texts_param(hrflow_client):
@@ -185,6 +188,7 @@ def test_tagger_rome_family_with_texts_param(hrflow_client):
     assert isinstance(model.data, list)
     assert len(model.data) == len(TAGGING_TEXTS)
 
+
 @pytest.mark.text
 @pytest.mark.tagging
 def test_tagger_rome_family_with_text_and_texts_param(hrflow_client):
@@ -198,9 +202,10 @@ def test_tagger_rome_family_with_text_and_texts_param(hrflow_client):
             )
         )
         pytest.fail("Should have raised a ValueError")
-    except ValueError as e:
+    except ValueError:
         pass
-    
+
+
 @pytest.mark.text
 @pytest.mark.tagging
 def test_tagger_rome_family_without_text_or_texts_param(hrflow_client):
@@ -212,9 +217,10 @@ def test_tagger_rome_family_without_text_or_texts_param(hrflow_client):
             )
         )
         pytest.fail("Should have raised a ValueError")
-    except ValueError as e:
+    except ValueError:
         pass
 
+
 def _tagging_test(
     hrflow_client: Hrflow,
     algorithm_key: TAGGING_ALGORITHM,
@@ -359,9 +365,7 @@ def test_ocr_basic(hrflow_client):
 495c951bbae6b/profiles/52e3c23a5f21190c59f53c41b5630ecb5d414f94/parsing/resume.pdf"""
     file = _file_get(s3_url, "ocr")
     assert file is not None
-    model = TextOCRResponse.model_validate(
-        hrflow_client.text.ocr.post(file=file)
-    )
+    model = TextOCRResponse.model_validate(hrflow_client.text.ocr.post(file=file))
     assert model.code == requests.codes.ok
     assert "ocr" in model.message.lower()
 

tests/utils/schemas.py

@@ -101,8 +101,7 @@ class TextTaggingDataItem(BaseModel):
     @classmethod
     def _check(cls, values: t.Dict[str, t.List[t.Any]]) -> t.Dict[str, t.List[t.Any]]:
         if isinstance(values, list):
-            return [cls._check(item) for item in values
-        ]
+            return [cls._check(item) for item in values]
         li = len(values.get("ids"))
         lp = len(values.get("predictions"))
         lt = len(values.get("tags"))