Update check_install.py

Author: C-Achard

dlclive/check_install/check_install.py

@@ -50,15 +50,12 @@ def run_pytorch_test(video_file: str, display: bool = False):
             f"Failed to export {TORCH_CONFIG['super_animal']} model"
         )
     if TORCH_CONFIG["checkpoint"].stat().st_size == 0:
-        raise ValueError(
-        f"Exported {TORCH_CONFIG['super_animal']} model is empty"
-        )
+        raise ValueError(f"Exported {TORCH_CONFIG['super_animal']} model is empty")
     benchmark_videos(
         model_path=str(TORCH_CONFIG["checkpoint"]),
         model_type="pytorch",
         video_path=video_file,
         display=display,
-        # resize=0.5,
         pcutoff=0.25,
         pixels=1000,
     )
@@ -88,15 +85,14 @@ def run_tensorflow_test(video_file: str, display: bool = False):
         model_type="base",
         video_path=video_file,
         display=display,
-        # resize=0.5,
         pcutoff=0.25,
         pixels=1000,
     )
 
 
 def main():
     backend_results = {}
-    
+
     parser = argparse.ArgumentParser(
         description="Test DLC-Live installation by downloading and evaluating a demo DLC project!"
     )
@@ -164,9 +160,11 @@ def main():
                     )
             except Exception as e:
                 backend_name = (
-                    "pytorch" if backend == Engine.PYTORCH else
-                    "tensorflow" if backend == Engine.TENSORFLOW else
-                    str(backend)
+                    "pytorch"
+                    if backend == Engine.PYTORCH
+                    else "tensorflow"
+                    if backend == Engine.TENSORFLOW
+                    else str(backend)
                 )
                 backend_results[backend_name] = ("ERROR", str(e))
                 backend_failures[backend] = e
@@ -191,7 +189,6 @@ def main():
             )
             raise RuntimeError(f"All backend tests failed. Details: {failure_messages}")
 
-
     finally:
         # deleting temporary files
         print("\n Deleting temporary files...\n")
@@ -204,7 +201,6 @@ def main():
             )
 
 
-
 if __name__ == "__main__":
     # Get available backends (emits a warning if neither TensorFlow nor PyTorch is installed)
     available_backends: list[Engine] = get_available_backends()