diff --git a/preprocessing/sports/event_data/soccer/soccer_processing.py b/preprocessing/sports/event_data/soccer/soccer_processing.py index ed2ae64..b5e43f8 100644 --- a/preprocessing/sports/event_data/soccer/soccer_processing.py +++ b/preprocessing/sports/event_data/soccer/soccer_processing.py @@ -1542,7 +1542,7 @@ def UIED_bepro(data): df["action"] = df["event_type"] = ( df["event_type"] .astype(str) - .str.replace("(Failed|Succeeded|Received|OnTarget)$", "", regex=True) + .str.replace("(Blocked|Failed|Succeeded|Received|OnTarget)$", "", regex=True) ) # Define possession team actions @@ -1662,7 +1662,7 @@ def UIED_bepro(data): action_list.append("dribble") elif df["action"].iloc[i] in cross_actions: action_list.append("cross") - elif df["action"].iloc[i] in drop_actions or np.isnan(df["action"].iloc[i]) or df["action"].iloc[i] in ["nan", "None"]: + elif df["action"].iloc[i] in drop_actions or pd.isna(df["action"].iloc[i]): action_list.append("drop") else: action= df["action"].iloc[i] diff --git a/pyproject.toml b/pyproject.toml index 4a63921..618a9db 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -6,7 +6,7 @@ build-backend = "setuptools.build_meta" [project] name = "openstarlab_preprocessing" -version = "0.1.46" +version = "0.1.47" description = "openstarlab preprocessing package" readme = "README.md" requires-python = ">=3.8"