From f65b1d194a8df4a28ebfda171df8ede6ba8c040f Mon Sep 17 00:00:00 2001 From: shunsuke-iwashita Date: Wed, 12 Nov 2025 14:53:51 +0900 Subject: [PATCH 1/2] Refactors imports and fixes preprocessing default Organizes import statements for readability and consistency, sets a default argument for preprocessing to improve usability, and cleans up redundant print statements for clarity. --- .../space_data/basketball/basketball_space_class.py | 13 +++++++------ 1 file changed, 7 insertions(+), 6 deletions(-) diff --git a/preprocessing/sports/space_data/basketball/basketball_space_class.py b/preprocessing/sports/space_data/basketball/basketball_space_class.py index 7119f2b..7203b78 100644 --- a/preprocessing/sports/space_data/basketball/basketball_space_class.py +++ b/preprocessing/sports/space_data/basketball/basketball_space_class.py @@ -1,11 +1,11 @@ import os +import zipfile + +import gdown import pandas as pd -from tqdm import tqdm from scipy.io import loadmat -import pandas as pd -import gdown -import zipfile -import os +from tqdm import tqdm + class Basketball_space_data: def __init__(self,data_provider,data_path, *args, **kwargs): @@ -13,7 +13,7 @@ def __init__(self,data_provider,data_path, *args, **kwargs): self.data_path = data_path - def preprocessing(self, nb_process_game): + def preprocessing(self, nb_process_game='ALL'): if self.data_provider == "SportVU_NBA": @@ -159,4 +159,5 @@ def download_data(self): os.remove(output_zip_path) print("delete the unecessary zip file") + return print("Data downloaded finish") print("delete the unecessary zip file") return print("Data downloaded finish") \ No newline at end of file From 94bd09d3ad95e2d664476f60bdd6cf964346e936 Mon Sep 17 00:00:00 2001 From: shunsuke-iwashita Date: Wed, 12 Nov 2025 14:54:04 +0900 Subject: [PATCH 2/2] Removes debug print from tracking data load Eliminates unnecessary console output to streamline data processing and improve log clarity during file handling. --- .../sports/space_data/ultimate/ultimate_space_class.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/preprocessing/sports/space_data/ultimate/ultimate_space_class.py b/preprocessing/sports/space_data/ultimate/ultimate_space_class.py index dbd5576..afa3f30 100644 --- a/preprocessing/sports/space_data/ultimate/ultimate_space_class.py +++ b/preprocessing/sports/space_data/ultimate/ultimate_space_class.py @@ -55,7 +55,7 @@ def preprocessing(self): os.path.splitext(os.path.basename(tracking_path_i))[0] )[0] match_tracking_df = pd.read_csv(tracking_path_i) - print(match_tracking_df) + # Create intermediate DataFrame with all required columns intermidiate_df = create_intermediate_file(match_tracking_df)