Reorganize module

.gitignore

@@ -0,0 +1,5 @@
+*.egg-info
+__pycache__
+*.py[cod]
+lightning_logs/
+.pytest_cache

submission.py → evaluate/submission.py

@@ -29,13 +29,3 @@ def make_submission(
             rows.append(row)
     submission_df = pd.DataFrame(rows)
     submission_df.to_csv(output_csv_path, index=False)
-
-
-if __name__ == "__main__":
-    with zipfile.ZipFile("/tmp/data/private_testset.zip", "r") as zip_ref:
-        zip_ref.extractall("/tmp/data")
-
-    MODEL_PATH = "last.ckpt"
-    metadata_file_path = "./SnakeCLEF2024-TestMetadata.csv"
-    test_metadata = pd.read_csv(metadata_file_path)
-    make_submission(test_metadata=test_metadata, model_path=MODEL_PATH)

test_evaluate.py → evaluate/test_evaluate.py

@@ -76,7 +76,7 @@ def test_model_checkpoint(model_checkpoint):
 def test_make_submission(model_checkpoint, metadata, images_root, tmp_path):
     output_csv_path = tmp_path / "submission.csv"
     make_submission(metadata, model_checkpoint, output_csv_path, images_root)
-    submission_df = pd.read_csv(output_csv_pathgit)
+    submission_df = pd.read_csv(output_csv_path)
     assert len(submission_df) == 10
     assert set(submission_df.columns) == {"observation_id", "class_id"}
     assert submission_df["class_id"].isin(range(10)).all()

script.py

@@ -0,0 +1,12 @@
+import zipfile
+import pandas as pd
+from evaluate.submission import make_submission
+
+if __name__ == "__main__":
+    with zipfile.ZipFile("/tmp/data/private_testset.zip", "r") as zip_ref:
+        zip_ref.extractall("/tmp/data")
+
+    MODEL_PATH = "./last.ckpt"
+    metadata_file_path = "./SnakeCLEF2024-TestMetadata.csv"
+    test_metadata = pd.read_csv(metadata_file_path)
+    make_submission(test_metadata=test_metadata, model_path=MODEL_PATH)