add typed function sig for run_detector

jac99 · Jul 4, 2023 · a8bb4a5 · a8bb4a5
1 parent 9e61877
commit a8bb4a5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/run_detector.py b/run_detector.py
@@ -39,7 +39,7 @@ def draw_bboxes(image, detections):
     return image
 
 
-def run_detector(model, args):
+def run_detector(model: footandball.FootAndBall, args: argparse.Namespace):
     model.print_summary(show_architecture=False)
     model = model.to(args.device)