Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add skyrim and dependencies #27273

Open
wants to merge 27 commits into
base: main
Choose a base branch
from
Open
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
80 changes: 80 additions & 0 deletions recipes/earth2mip/meta.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1,80 @@
{% set name = "earth2mip" %}
{% set version = "0.1.0" %}

package:
name: {{ name|lower }}
version: {{ version }}

source:
url: https://github.com/NVIDIA/earth2mip/archive/refs/tags/v{{ version }}.tar.gz
sha256: a49d0607893013783d30bfcb2f80412014ab535fbcc1e96dd139b78819bd98ab

build:
noarch: python
script:
- sed -i.bak '/"pytest/d' pyproject.toml
- {{ PYTHON }} -m pip install . -vv --no-deps --no-build-isolation
number: 0

requirements:
host:
- python >=3.10
- setuptools
- setuptools_scm
- pip
run:
- python >=3.10
- altair >=4.2.2
- cdsapi >=0.6.1
- cfgrib >=0.9.10.3
- cftime
- dask >=2023.1.0
- distributed >=2023.1.0
- eccodes >=1.4.0
- ecmwflibs >=0.5.2
- einops
- fsspec
- h5py >=3.2.0
- h5netcdf >=1.0.0
- importlib-metadata >=6.7.0
- joblib >=1.1.0
- netcdf4 >=1.6.4
- numpy
- nvidia-modulus >=0.4.0
- pandas >=1.5.3
- properscoring >=0.1
- pydantic >=1.10, <=1.10.11
- python-dotenv >=1.0.0
- s3fs
- setuptools >=38.4
- pytorch >=1.13
- torch-harmonics >=0.5.0
- tqdm >=4.65.0
- typer
- xarray
- xskillscore >=0.0.24
- zarr >=2.14.2

test:
imports:
- earth2mip
commands:
- pip check
requires:
- pip

about:
home: https://nvidia.github.io/earth2mip
dev_url: https://github.com/NVIDIA/earth2mip
summary: |
Earth-2 Model Intercomparison Project (MIP) is a python framework that
enables climate researchers and scientists to explore and experiment
with AI models for weather and climate.
license: Apache-2.0
license_file:
- LICENSE.txt

extra:
recipe-maintainers:
- isuruf
- timkpaine
62 changes: 62 additions & 0 deletions recipes/nvidia-modulus/meta.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1,62 @@
{% set name = "nvidia-modulus" %}
{% set version = "0.7.0" %}

package:
name: {{ name|lower }}
version: {{ version }}

source:
url: https://github.com/NVIDIA/modulus/archive/refs/tags/v0.7.0.tar.gz
sha256: 8c81b53b1b106e9af9a60c2ce416a93f2e44ebc00061fb7d0631ccb63f8f0ae8

build:
noarch: python
script:
- sed -i.bak '/nvidia_dali_cuda120/d' pyproject.toml
- {{ PYTHON }} -m pip install . -vv --no-deps --no-build-isolation
number: 0

requirements:
host:
- python >=3.10
- setuptools
- setuptools-scm
- pip
run:
- python >=3.10
- pytorch >=2.0.0
- numpy >=1.22.4,<1.25
- xarray >=2023.1.0
- zarr >=2.14.2
- fsspec >=2023.1.0
- s3fs >=2023.5.0
- nvidia-dali-python >=1.35.0
- setuptools >=67.6.0
- certifi >=2023.7.22
- pytz >=2023.3
- treelib >=1.2.5
- tqdm >=4.60.0
- nvtx >=0.2.8
- onnx >=1.14.0
- timm >=0.9.12
- importlib-metadata

test:
imports:
- modulus
commands:
- pip check
requires:
- pip

about:
home: https://github.com/NVIDIA/modulus
summary: A deep learning framework for AI-driven multi-physics systems
license: Apache-2.0
license_file:
- LICENSE.txt

extra:
recipe-maintainers:
- isuruf
- timkpaine
61 changes: 61 additions & 0 deletions recipes/skyrim/meta.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1,61 @@
{% set name = "skyrim" %}
{% set version = "0.0.2" %}

package:
name: {{ name|lower }}
version: {{ version }}

source:
url: https://pypi.io/packages/source/{{ name[0] }}/{{ name }}/skyrim-{{ version }}.tar.gz
sha256: 53d1a19b9ddd6a1aef69812b092d392fc09cd735a2220ded245d46a8735cae35

build:
entry_points:
- forecast = skyrim.forecast:main
noarch: python
script:
- sed -i.bak "s/s3fs==0.4.2/s3fs>=0.4.2/g" pyproject.toml
- {{ PYTHON }} -m pip install . -vv --no-deps --no-build-isolation
number: 0

requirements:
host:
- python >=3.10
- setuptools
- setuptools-scm
- pip
run:
- python >=3.10
- modal
- loguru
- xarray
- base58
- botocore
- aiobotocore
- boto3
- s3fs >=0.4.2
- zarr
- scipy
- pandas
- click
- earth2mip 0.1.*

test:
imports:
- skyrim
commands:
- pip check
- forecast --help
requires:
- pip

about:
home: https://github.com/secondlaw-ai/skyrim
summary: AI weather models united.
license: Apache-2.0
license_file: LICENSE

extra:
recipe-maintainers:
- isuruf
- timkpaine
Loading