fix versioning

.github/workflows/pytest.yml

@@ -4,6 +4,7 @@ on:
   push:
     branches:
       - main
+      - bug/crystal_conv_saving
   pull_request:
     branches:
       - main
@@ -13,7 +14,7 @@ jobs:
     strategy:
       matrix:
         os: [ubuntu-latest, windows-latest, macos-latest]
-        python-version: ['3.10', '3.11', '3.12']
+        python-version: ['3.11']
 
     runs-on: ${{ matrix.os }}
 

.gitignore

@@ -167,6 +167,7 @@ cython_debug/
 /.venv
 /.pytest_cache
 .DS_Store
+.venv*
 
 dev.py
 bug.py
@@ -179,5 +180,9 @@ build.py
 /pickle_files
 **/pickle_files
 
-tut.ipynb
+*.ipynb
+*.keras
+*.pkl
+*.txt
 
+/models

requirements.txt

@@ -1,4 +1,5 @@
-spektral==1.3.1
-kloppy @ git+https://github.com/PySport/kloppy.git@9ccbc77c57c2caafaaf04f5aaf090deea8f03b7d
-pytest==8.2.2
-"black[jupyter]==24.4.2"
+spektral==1.2.0
+git+https://github.com/PySport/kloppy.git@9ccbc77c57c2caafaaf04f5aaf090deea8f03b7d
+tensorflow>=2.14.0; platform_machine != 'arm64' or platform_system != 'Darwin'
+tensorflow-macos>=2.14.0; platform_machine == 'arm64' and platform_system == 'Darwin'
+keras==2.14.0

setup.py

@@ -11,16 +11,17 @@
     url="https://github.com/unravelsports/unravelsports",
     packages=["unravel"] + ["unravel." + pkg for pkg in find_packages("unravel")],
     classifiers=[
-        "Programming Language :: Python :: 3.10",
         "Programming Language :: Python :: 3.11",
-        "Programming Language :: Python :: 3.12",
         "License :: OSI Approved :: GNU Affero General Public License version 3",
         "Operating System :: OS Independent",
     ],
-    python_requires=">=3.10",
+    python_requires="~=3.11",
     install_requires=[
-        "spektral==1.3.1",
+        "spektral==1.2.0",
         "kloppy @ git+https://github.com/PySport/kloppy.git@9ccbc77c57c2caafaaf04f5aaf090deea8f03b7d",
+        "tensorflow>=2.14.0;platform_machine != 'arm64' or platform_system != 'Darwin'",
+        "tensorflow-macos>=2.14.0;platform_machine == 'arm64' and platform_system == 'Darwin'",
+        "keras==2.14.0",
     ],
     extras_require={
         "test": [

unravel/utils/objects/custom_spektral_dataset.py

@@ -223,11 +223,3 @@ def __handle_graph_id(i):
                 return self[train_idxs], self[test_idxs], self[validation_idxs]
             else:
                 return self[train_idxs], self[test_idxs]
-
-    def get_config(self):
-        config = {"kwargs": self._kwargs}
-        return config
-
-    @classmethod
-    def from_config(cls, config):
-        return cls(**config["kwargs"])