fix black formatting

nestauk · Jul 9, 2024 · 0f9e570 · 0f9e570
1 parent 2d3ea01
commit 0f9e570
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 4 deletions.
diff --git a/ojd_daps_skills/extract_skills/extract_skills.py b/ojd_daps_skills/extract_skills/extract_skills.py
@@ -73,7 +73,7 @@ def extract_skills(self, job_ads: Union[str, List[str]]) -> List[Doc]:
 
     # map skills function
 
-    def get_skills(self, job_ad: str, min_length: int=75) -> Doc:
+    def get_skills(self, job_ad: str, min_length: int = 75) -> Doc:
         """Return a spaCy Doc object with entities
             and split 'SKILL' spans.
 
@@ -103,9 +103,11 @@ def get_skills(self, job_ad: str, min_length: int=75) -> Doc:
                             for rule in rules:
                                 split_ent = rule(ent)
                                 if split_ent:
-                                    all_skill_ents += split_ent # Add the list of split skills
+                                    all_skill_ents += (
+                                        split_ent  # Add the list of split skills
+                                    )
                                     split_found = True
-                                    break # stop going through rules
+                                    break  # stop going through rules
                         if not split_found:
                             # else, if no split, append the original entity
                             all_skill_ents.append(ent)

diff --git a/ojd_daps_skills/utils/download_public_data.py b/ojd_daps_skills/utils/download_public_data.py
@@ -20,7 +20,9 @@ def download_data():
     key = "escoe_extension/ojd_daps_skills_data_refactor.zip"
 
     try:
-        s3.download_file(bucket_name, key, f"{str(PUBLIC_DATA_FOLDER_PATH)}_refactor.zip")
+        s3.download_file(
+            bucket_name, key, f"{str(PUBLIC_DATA_FOLDER_PATH)}_refactor.zip"
+        )
 
         with ZipFile(f"{PUBLIC_DATA_FOLDER_PATH}_refactor.zip", "r") as zip_ref:
             zip_ref.extractall(PROJECT_DIR)