bris-forecaster / pyproject.toml
evenmn
Added training config, pyproject.toml as well as better instructions in README.md
9f28f00
Raw
History Blame Contribute Delete
5.09 kB
[project]
name = "calm-coriolis-bris-inference"
version = "0.1.0"
description = "uv project for the calm-coriolis bris-inference environment"
requires-python = "==3.12.11"
dependencies = [
"aiohappyeyeballs==2.6.1",
"aiohttp==3.12.13",
"aiosignal==1.3.2",
"alembic==1.16.3",
"anemoi-datasets==0.5.24",
"anemoi-graphs==0.6.2",
"anemoi-models==0.8.1",
"anemoi-training==0.5.1",
"anemoi-transform==0.1.11",
"anemoi-utils==0.4.23",
"aniso8601==10.0.1",
"annotated-types==0.7.0",
"antlr4-python3-runtime==4.9.3",
"anyio==4.9.0",
"array-api-compat==1.12.0",
"asciitree==0.3.3",
"astropy==7.1.0",
"astropy-iers-data==0.2025.6.30.0.39.40",
"attrs==25.3.0",
"blinker==1.9.0",
"bris",
"cachetools==5.5.2",
"certifi==2025.6.15",
"cffi==1.17.1",
"cfgrib==0.9.15.0",
"cftime==1.6.4.post1",
"cfunits==3.3.7",
"charset-normalizer==3.4.2",
"click==8.2.1",
"cloudpickle==3.1.1",
"colorcet==3.1.0",
"contourpy==1.3.2",
"cycler==0.12.1",
"dask==2025.5.1",
"databricks-sdk==0.58.0",
"datashader==0.18.1",
"Deprecated==1.2.18",
"deprecation==2.1.0",
"docker==7.1.0",
"earthkit-data==0.15.0",
"earthkit-meteo==0.4.1",
"earthkit-regrid==0.4.0",
"earthkit-utils==0.0.1",
"eccodes==2.42.0",
"einops==0.8.1",
"entrypoints==0.4",
"fastapi==0.116.0",
"fasteners==0.19",
"filelock==3.18.0",
"findlibs==0.1.1",
"Flask==3.1.1",
"flexcache==0.3",
"flexparser==0.4",
"fonttools==4.58.4",
"frozenlist==1.7.0",
"fsspec==2025.5.1",
"future==1.0.0",
"gitdb==4.0.12",
"GitPython==3.1.44",
"google-auth==2.40.3",
"graphene==3.4.3",
"graphql-core==3.2.6",
"graphql-relay==3.2.0",
"greenlet==3.2.3",
"gridpp==0.7.1",
"gunicorn==23.0.0",
"h11==0.16.0",
"h3==3.7.7",
"healpy==1.18.1",
"hf-xet==1.1.5",
"huggingface-hub==0.33.2",
"hydra-core==1.3.2",
"idna==3.10",
"importlib_metadata==8.7.0",
"itsdangerous==2.2.0",
"Jinja2==3.1.6",
"joblib==1.5.1",
"jsonschema==4.24.0",
"jsonschema-specifications==2025.4.1",
"kiwisolver==1.4.8",
"lightning-utilities==0.14.3",
"llvmlite==0.44.0",
"locket==1.0.0",
"lru-dict==1.3.0",
"Mako==1.3.10",
"Markdown==3.8.2",
"MarkupSafe==3.0.2",
"matplotlib==3.10.3",
"mlflow==3.1.1",
"mlflow-skinny==3.1.1",
"mpmath==1.3.0",
"multidict==6.6.3",
"multipledispatch==1.0.0",
"multiurl==0.3.5",
"narwhals==1.44.0",
"netCDF4==1.7.2",
"networkx==3.5",
"numba==0.61.2",
"numcodecs==0.15.1",
"numpy==1.26.4",
"nvidia-cublas-cu12==12.4.5.8",
"nvidia-cuda-cupti-cu12==12.4.127",
"nvidia-cuda-nvrtc-cu12==12.4.127",
"nvidia-cuda-runtime-cu12==12.4.127",
"nvidia-cudnn-cu12==9.1.0.70",
"nvidia-cufft-cu12==11.2.1.3",
"nvidia-cufile-cu12==1.11.1.6",
"nvidia-curand-cu12==10.3.5.147",
"nvidia-cusolver-cu12==11.6.1.9",
"nvidia-cusparse-cu12==12.3.1.170",
"nvidia-cusparselt-cu12==0.6.2",
"nvidia-ml-py==12.575.51",
"nvidia-nccl-cu12==2.21.5",
"nvidia-nvjitlink-cu12==12.4.127",
"nvidia-nvtx-cu12==12.4.127",
"nvsmi==0.4.2",
"omegaconf==2.3.0",
"opentelemetry-api==1.34.1",
"opentelemetry-sdk==1.34.1",
"opentelemetry-semantic-conventions==0.55b1",
"packaging==25.0",
"pandas==2.3.0",
"param==2.2.1",
"partd==1.4.2",
"pdbufr==0.14.0",
"pillow==11.3.0",
"Pint==0.24.4",
"platformdirs==4.3.8",
"plotly==6.2.0",
"pooch==1.8.2",
"propcache==0.3.2",
"protobuf==6.31.1",
"psutil==7.0.0",
"pyarrow==20.0.0",
"pyasn1==0.6.1",
"pyasn1_modules==0.4.2",
"pycparser==2.22",
"pyct==0.5.0",
"pydantic==2.11.7",
"pydantic_core==2.33.2",
"pyerfa==2.0.1.5",
"pynvml==12.0.0",
"pyparsing==3.2.3",
"pyproj==3.7.1",
"pyshtools==4.13.1",
"python-dateutil==2.9.0.post0",
"pytorch-lightning==2.5.2",
"pytz==2025.2",
"PyYAML==6.0.2",
"referencing==0.36.2",
"requests==2.32.4",
"rpds-py==0.25.1",
"rsa==4.9.1",
"ruff==0.12.7",
"safetensors==0.5.3",
"scikit-learn==1.7.0",
"scipy==1.15.3",
"semantic-version==2.10.0",
"six==1.17.0",
"smmap==5.0.2",
"sniffio==1.3.1",
"SQLAlchemy==2.0.41",
"sqlparse==0.5.3",
"starlette==0.46.2",
"sympy==1.13.1",
"threadpoolctl==3.6.0",
"timm==1.0.16",
"toolz==1.0.0",
"torch==2.6.0+cu124",
"torch-geometric==2.6.1",
"torchaudio==2.6.0+cu124",
"torchinfo==1.8.0",
"torchmetrics==1.7.3",
"torchvision==0.21.0+cu124",
"tqdm==4.67.1",
"trimesh==4.6.13",
"triton==3.2.0",
"typeguard==4.4.4",
"typing-inspection==0.4.1",
"typing_extensions==4.14.0",
"tzdata==2025.2",
"urllib3==2.5.0",
"uvicorn==0.35.0",
"verif==1.3.0",
"Werkzeug==3.1.3",
"wrapt==1.17.2",
"xarray==2025.6.1",
"yarl==1.20.1",
"zarr==2.17.0",
"zipp==3.23.0",
]
[tool.uv]
index-strategy = "unsafe-best-match"
[[tool.uv.index]]
name = "pytorch-cu124"
url = "https://download.pytorch.org/whl/cu124"
explicit = true
[tool.uv.sources]
bris = { git = "ssh://git@github.com/metno/bris-inference.git", rev = "d1d27c11aef811d0cecaa5e6134ae0f41f4ebb76" }
torch = { index = "pytorch-cu124" }
torchaudio = { index = "pytorch-cu124" }
torchvision = { index = "pytorch-cu124" }