From a8bb4a5a5105d2face1e6b25448a7ac5ba1eafa6 Mon Sep 17 00:00:00 2001 From: Jayandjeff Date: Tue, 4 Jul 2023 11:32:13 -0700 Subject: [PATCH] add typed function sig for run_detector --- run_detector.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/run_detector.py b/run_detector.py index 0fcb74e..0405e01 100644 --- 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)