[project]
name = "entropice"
version = "0.1.0"
description = "Add your description here"
readme = "README.md"
authors = [{ name = "Tobias Hölzer", email = "tobiashoelzer@hotmail.com" }]
# requires-python = ">=3.10,<3.13"
dependencies = [
    "aiohttp>=3.12.11",
    "bokeh>=3.7.3",
    "bottleneck>=1.6.0",
    "cartopy>=0.24.1",
    "cdsapi>=0.7.6",
    "cyclopts>=4.0.0",
    "dask>=2025.5.1",
    "distributed>=2025.5.1",
    "earthengine-api>=1.6.9",
    "eemont>=2025.7.1",
    "entropy",
    "flox>=0.10.4",
    "folium>=0.19.7",
    "geemap>=0.36.3",
    "geopandas>=1.1.0",
    "h3>=4.2.2",
    "ipycytoscape>=1.3.3",
    "ipykernel>=6.29.5",
    "ipywidgets>=8.1.7",
    "mapclassify>=2.9.0",
    "matplotlib>=3.10.3",
    "netcdf4>=1.7.2",
    "numba>=0.62.1",
    "numbagg>=0.9.3",
    "numpy>=2.3.0",
    "odc-geo[all]>=0.4.10",
    "opt-einsum>=3.4.0",
    "pyarrow>=20.0.0",
    "rechunker>=0.5.2",
    "requests>=2.32.3",
    "rich>=14.0.0",
    "rioxarray>=0.19.0",
    "scipy>=1.15.3",
    "seaborn>=0.13.2",
    "smart-geocubes[gee,dask,stac,viz]>=0.0.9",
    "stopuhr>=0.0.10",
    "ultraplot>=1.63.0",
    "xanimate",
    "xarray>=2025.9.0",
    "xdggs>=0.2.1",
    "xvec>=0.5.1",
    "zarr[remote]>=3.1.3",
    "geocube>=0.7.1,<0.8",
    "streamlit>=1.50.0,<2",
    "altair[all]>=5.5.0,<6",
    "h5netcdf>=1.7.3,<2", "streamlit-folium>=0.25.3,<0.26", "st-theme>=1.2.3,<2",
]

[project.scripts]
create-grid = "entropice.grids:main"
darts = "entropice.darts:main"
alpha-earth = "entropice.alphaearth:main"
era5 = "entropice.era5:cli"
train = "entropice.training:main"
dataset = "entropice.dataset:main"

[build-system]
requires = ["hatchling"]
build-backend = "hatchling.build"

[tool.uv]
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]
dashboard = { cmd = [
    "streamlit",
    "run",
    "src/entropice/training_analysis_dashboard.py",
    "--server.port",
    "8501",
    "--server.address",
    "0.0.0.0",
] }

[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.*"