From f463e390ff5cd5e4e1493157d7cd7f3745b4e6e8 Mon Sep 17 00:00:00 2001 From: Calvin Yeung <70192523+calvinyeungck@users.noreply.github.com> Date: Wed, 8 Oct 2025 14:32:00 +0900 Subject: [PATCH] Move imports for tracking data classes inside __new__ --- preprocessing/sports/tracking_data/tracking_class.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/preprocessing/sports/tracking_data/tracking_class.py b/preprocessing/sports/tracking_data/tracking_class.py index ff6ac95..cd009d8 100644 --- a/preprocessing/sports/tracking_data/tracking_class.py +++ b/preprocessing/sports/tracking_data/tracking_class.py @@ -1,7 +1,3 @@ -from .soccer.soccer_tracking_class import Soccer_tracking_data -from .ultimate.ultimate_tracking_class import Ultimate_tracking_data - - class Tracking_data: soccer_data_provider = ["soccer"] ultimate_data_provider = ["UltimateTrack", "UFA"] @@ -10,8 +6,10 @@ class Tracking_data: def __new__(cls, data_provider, *args, **kwargs): if data_provider in cls.soccer_data_provider: + from .soccer.soccer_tracking_class import Soccer_tracking_data return Soccer_tracking_data(*args, **kwargs) elif data_provider in cls.ultimate_data_provider: + from .ultimate.ultimate_tracking_class import Ultimate_tracking_data return Ultimate_tracking_data(*args, **kwargs) elif data_provider in cls.handball_data_provider: raise NotImplementedError("Handball event data not implemented yet")