flake8 compliant

tests/test_generate_random_prediction.py

@@ -10,8 +10,12 @@ if __name__ == "__main__":
     img_path_list = [path for path in img_dir.glob("*.jpeg") if path.is_file()]
     print(img_path_list)
     # instantiate data object
-    vis_com_list = [VisualCommunication.from_file(path) for path in img_path_list]
+    vis_com_list = [
+        VisualCommunication.from_file(path)
+        for path
+        in img_path_list
+    ]
     # generate random predictions
     [vis_com.generate_random_prediction() for vis_com in vis_com_list]
     for vis_com in vis_com_list:
-        print(vis_com)
+        print(vis_com)

tests/test_image_upload.py

@@ -12,7 +12,11 @@ if __name__ == "__main__":
     img_path_list = [path for path in img_dir.glob("*.jpeg") if path.is_file()]
     print(img_path_list)
     # instantiate data object
-    vis_com_list = [VisualCommunication.from_file(path) for path in img_path_list]
+    vis_com_list = [
+        VisualCommunication.from_file(path)
+        for path
+        in img_path_list
+    ]
     for vis_com in vis_com_list:
         print(repr(vis_com))
     # prepare env vars

tests/test_model_outputs_from_annotation.py

@@ -3,11 +3,12 @@ from src.database import ModelOutputs
 
 if __name__ == "__main__":
     # instantiate data object
-    annotation = {
-        
-    }
-    vis_com_list = [ModelOutputs.from_annotation(path) for path in img_path_list]
+    vis_com_list = [
+        ModelOutputs.from_random()
+        for i
+        in range(3)
+    ]
     # generate random predictions
     [vis_com.generate_random_prediction() for vis_com in vis_com_list]
     for vis_com in vis_com_list:
-        print(vis_com)
+        print(vis_com)

tests/test_prediction_upload.py

@@ -25,7 +25,11 @@ if __name__ == "__main__":
     img_dir = test_dir / "imgs"
     img_path_list = [path for path in img_dir.glob("*.jpeg") if path.is_file()]
     # instantiate data object
-    vis_com_list = [VisualCommunication.from_file(path) for path in img_path_list]
+    vis_com_list = [
+        VisualCommunication.from_file(path)
+        for path
+        in img_path_list
+    ]
     # generate random predictions
     [vis_com.generate_random_prediction() for vis_com in vis_com_list]
     # prepare env vars

tests/test_total_annotated.py

@@ -17,4 +17,4 @@ if __name__ == "__main__":
     collection, db, client = connect()
     # get visual communication
     num_docs = total_annotated(collection)
-    print(f"number of annotated  documents in database: {num_docs}")
+    print(f"number of annotated  documents in database: {num_docs}")

tests/test_total_documents.py

@@ -17,4 +17,4 @@ if __name__ == "__main__":
     collection, db, client = connect()
     # get visual communication
     num_docs = total_documents(collection)
-    print(f"total number of documents in database: {num_docs}")
+    print(f"total number of documents in database: {num_docs}")