Skip to content

Commit

Permalink
chore: Update pylint workflow to include src directory
Browse files Browse the repository at this point in the history
in test coverage

The pytest command now includes the src directory
for test coverage, ensuring that all codebase tests are accounted for.
  • Loading branch information
aus10powell committed Sep 14, 2024
1 parent 5315472 commit 15425b3
Showing 1 changed file with 22 additions and 0 deletions.
22 changes: 22 additions & 0 deletions code/run_pipeline.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,22 @@
import sys, os
from pathlib import Path

# Add the src directory to the Python path
src_path = Path(__file__).resolve().parent / 'src'
sys.path.insert(0, str(src_path))

from pipeline import main # Assuming you have a main function in pipeline.py

if __name__ == "__main__":
OUTPUT_DIR = os.path.join("/", "Users", "aus10powell", "Downloads")
video_path = "/Users/aus10powell/Documents/Projects/MIT-Fishery-Counter/data/gold_dataset/videos/irwa/1_2016-04-22_12-36-58.mp4"
tracker = "/Users/aus10powell/Documents/Projects/MIT-Fishery-Counter/code/src/utils/tracking_configs/botsort.yaml"
model_path = "/Users/aus10powell/Documents/Projects/MIT-Fishery-Counter/code/notebooks/runs/detect/train133/weights/best.pt"
params = {
"video_path": video_path,
"OUTPUT_DIR": OUTPUT_DIR,
"tracker": tracker,
"model_path": model_path,
"write_to_local": True
}
main(params)

0 comments on commit 15425b3

Please sign in to comment.