Add first training

pyproject.toml

@@ -4,7 +4,7 @@ version = "0.1.0"
 description = "Add your description here"
 readme = "README.md"
 authors = [{ name = "Tobias Hölzer", email = "tobiashoelzer@hotmail.com" }]
-requires-python = ">=3.12"
+# requires-python = ">=3.10,<3.13"
 dependencies = [
     "aiohttp>=3.12.11",
     "bokeh>=3.7.3",
@@ -16,7 +16,7 @@ dependencies = [
     "distributed>=2025.5.1",
     "earthengine-api>=1.6.9",
     "eemont>=2025.7.1",
-    "entropyc",
+    "entropy",
     "flox>=0.10.4",
     "folium>=0.19.7",
     "geemap>=0.36.3",
@@ -48,7 +48,7 @@ dependencies = [
     "xarray>=2025.9.0",
     "xdggs>=0.2.1",
     "xvec>=0.5.1",
-    "zarr[remote]>=3.1.3",
+    "zarr[remote]>=3.1.3", "geocube>=0.7.1,<0.8",
 ]
 
 [project.scripts]
@@ -56,6 +56,7 @@ create-grid = "entropice.grids:main"
 darts = "entropice.darts:main"
 alpha-earth = "entropice.alphaearth:main"
 era5 = "entropice.era5:cli"
+train = "entropice.training:cli"
 
 [build-system]
 requires = ["hatchling"]
@@ -66,4 +67,40 @@ package = true
 
 [tool.uv.sources]
 entropyc = { git = "ssh://git@github.com/AlbertEMC2Stein/entropyc", branch = "refactor/tobi" }
+entropy = { git = "ssh://git@forgejo.tobiashoelzer.de:22222/tobias/entropy.git" }
 xanimate = { git = "https://github.com/davbyr/xAnimate" }
+
+[tool.ruff.lint.pyflakes]
+# Ignore libraries when checking for unused imports
+allowed-unused-imports = [
+    "hvplot.pandas",
+    "hvplot.xarray",
+    "rioxarray",
+    "odc.geo.xr",
+    "cupy_xarray",
+    "xdggs",
+    "xvec",
+]
+
+[tool.pixi.workspace]
+channels = ["conda-forge"]
+platforms = ["linux-64"]
+
+[tool.pixi.activation.env]
+SCIPY_ARRAY_API = "1"
+
+[tool.pixi.system-requirements]
+cuda = "12"
+
+[tool.pixi.pypi-dependencies]
+entropice = { path = ".", editable = true }
+
+[tool.pixi.tasks]
+
+[tool.pixi.dependencies]
+pytorch-gpu = ">=2.5.1,<3"
+cupy = ">=13.6.0,<14"
+nccl = ">=2.27.7.1,<3"
+cudnn = ">=9.13.1.26,<10"
+cusparselt = ">=0.8.1.1,<0.9"
+cuda-version = "12.1.*"