black formatting

Author: lukestroh

pybullet_tree_sim/utils/mesh_downloader.py

@@ -11,24 +11,25 @@
 
 
 def get_filename_from_url(url: str) -> str:
-    local_filename = url.split('/')[-1]
+    local_filename = url.split("/")[-1]
     return os.path.join(meshes_path, local_filename)
 
 
 def is_download_needed(target_file_path: str) -> bool:
     if not os.path.exists(meshes_path):
         os.mkdir(meshes_path)
         return True
-    if not os.path.exists(os.path.join(meshes_path, 'trees')):
-        if not os.path.isfile(os.path.join(meshes_path, 'pybullet-tree-sim-meshes.zip')): # TODO: pass name into func
+    if not os.path.exists(os.path.join(meshes_path, "trees")):
+        if not os.path.isfile(os.path.join(meshes_path, "pybullet-tree-sim-meshes.zip")):  # TODO: pass name into func
             return True
         else:
             return False
     else:
         return False
 
+
 def is_unzip_needed(target_file_path: str) -> bool:
-    if not os.path.exists(os.path.join(meshes_path, 'trees')):
+    if not os.path.exists(os.path.join(meshes_path, "trees")):
         return True
     else:
         return False
@@ -40,25 +41,24 @@ def download_file(url: str, target_file_path: str) -> bool:
         with requests.get(url, stream=True) as r:
             r.raise_for_status()
             total_size = int(r.headers.get("content-length", 0))
-            with tqdm(total=total_size, unit='B', unit_scale=True) as progress_bar:
-                with open(target_file_path, 'wb') as f:
-                    for chunk in r.iter_content(chunk_size=8192): 
+            with tqdm(total=total_size, unit="B", unit_scale=True) as progress_bar:
+                with open(target_file_path, "wb") as f:
+                    for chunk in r.iter_content(chunk_size=8192):
                         # If you have chunk encoded response uncomment if
                         # and set chunk_size parameter to None.
-                        # if chunk: 
+                        # if chunk:
                         progress_bar.update(len(chunk))
                         f.write(chunk)
-        
+
                     return True
-    
+
     except Exception as e:
-        print(f'{e}')
+        print(f"{e}")
         return False
-    
 
 
 def unzip(zip_file: str):
-    with zipfile.ZipFile(zip_file, 'r') as zipper:
+    with zipfile.ZipFile(zip_file, "r") as zipper:
         zipper.extractall(os.path.dirname(zip_file))
     return
 
@@ -71,7 +71,7 @@ def main():
     download_needed = is_download_needed(target_file_path=file_abs_path)
     if download_needed:
         download_file(url=url, target_file_path=file_abs_path)
-    
+
     unzip_needed = is_unzip_needed(target_file_path=file_abs_path)
     if unzip_needed:
         unzip(zip_file=file_abs_path)