Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
10 changes: 6 additions & 4 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -15,9 +15,11 @@
The **unravelsports** package aims to aid researchers, analysts and enthusiasts by providing intermediary steps in the complex process of converting raw sports data into meaningful information and actionable insights.

This package currently supports:
- ⚽🏈 [**Polars DataFrame Conversion**](#polars-dataframes)
- ⚽🏈 [**Graph Neural Network**](#graph-neural-networks) Training, Graph Conversion and Prediction
- ⚽ [**Pressing Intensity**](#pressing-intensity) [[Bekkers (2024)](https://unravelsports.github.io/2024/12/12/pressing-intensity.html)]
- ⚽ 🏈 [**Polars DataFrame Conversion**](#polars-dataframes)
- ⚽ 🏈 [**Graph Neural Network**](#graph-neural-networks) Training, Graph Conversion and Prediction <small>
[[Bekkers & Sahasrabudhe (2023)](https://arxiv.org/pdf/2411.17450)]</small>
- ⚽ [**Pressing Intensity**](#pressing-intensity)
<small>[[Bekkers (2024)](https://arxiv.org/pdf/2501.04712)]</small>

🌀 Features
-----
Expand Down Expand Up @@ -138,7 +140,7 @@ The easiest way to get started is:
pip install unravelsports
```

Due to compatibility issues **unravelsports** currently only works on Python 3.11 with:
⚠️ Due to compatibility issues **unravelsports** currently only works on Python 3.11 with:
```
spektral==1.20.0
tensorflow==2.14.0
Expand Down
4 changes: 3 additions & 1 deletion unravel/soccer/dataset/kloppy_polars.py
Original file line number Diff line number Diff line change
Expand Up @@ -405,7 +405,9 @@ def __melt(
for k, coordinate in enumerate([Column.X, Column.Y, Column.Z]):
if object.id != Constant.BALL and coordinate == Column.Z:
continue
if not any(object.id in column for column in columns):
if not any(
object.id + "_" + coordinate == column for column in columns
):
continue

melted_df = self.__unpivot(df, object, coordinate)
Expand Down