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 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)