diff --git a/easybuild/easyconfigs/a/Anemoi-Framework/Anemoi-Framework-2025.06-foss-2024a.eb b/easybuild/easyconfigs/a/Anemoi-Framework/Anemoi-Framework-2025.06-foss-2024a.eb new file mode 100644 index 000000000000..b9edfcba6f5f --- /dev/null +++ b/easybuild/easyconfigs/a/Anemoi-Framework/Anemoi-Framework-2025.06-foss-2024a.eb @@ -0,0 +1,94 @@ +easyblock = 'PythonBundle' + +name = 'Anemoi-Framework' +version = '2025.06' + +homepage = 'https://anemoi.readthedocs.io' +description = """The Anemoi framework provides a complete toolkit to develop data-driven weather +models – from data preparation through to inference. The framework is composed +of several packages which target the different components necessary to +construct data-driven weather models. To aid development and deployment, each +package collects metadata that can be used by the subsequent packages. The +framework builds upon on established Python tools including PyTorch, Lighting, +Hydra, Zarr, Xarray and earthkit.""" + +toolchain = {'name': 'foss', 'version': '2024a'} + +dependencies = [ + ('Python', '3.12.3'), + ('SciPy-bundle', '2024.05'), + ('PyTorch-Geometric', '2.6.1'), + ('cfunits', '3.3.7'), + ('earthkit', '0.10.3'), + ('einops', '0.8.1'), + ('MLflow', '2.22.4'), + ('pyshtools', '4.13.1'), + ('timm', '1.0.20'), + ('zarr', '2.18.4'), +] + +exts_default_options = { + 'source_tmpl': SOURCE_PY3_WHL, +} + +exts_list = [ + ('aniso8601', '10.0.1', { + 'source_tmpl': SOURCE_WHL, + 'checksums': ['eb19717fd4e0db6de1aab06f12450ab92144246b257423fe020af5748c0cb89e'], + }), + ('anemoi_utils', '0.4.27', { + 'modulename': 'anemoi.utils', + 'checksums': ['f3c2d3d0b2c623d5db619733680edc9422fe5a1e179c7328f3b4369c5bdbbc20'], + }), + ('anemoi_datasets', '0.5.25', { + 'modulename': 'anemoi.datasets', + 'checksums': ['115290d630bf3a955a38d6dcbab9c8d8f9dfcdac05a708fe892cf3aab4192f49'], + }), + ('typeguard', '4.4.2', { + 'checksums': ['77a78f11f09777aeae7fa08585f33b5f4ef0e7335af40005b0c422ed398ff48c'], + }), + ('anemoi_graphs', '0.6.2', { + 'modulename': 'anemoi.graphs', + 'checksums': ['543b065b06c1ae077efdefef9770fcfb369ee0bbd7f3ac928c472f6b07cf4772'], + }), + ('anemoi_models', '0.8.1', { + 'modulename': 'anemoi.models', + 'checksums': ['fb1686775120aeb6c05164d5bcbe2b31f0269b4107da216fcc9c6f53e5fb518e'], + }), + ('anytree', '2.13.0', { + 'checksums': ['4cbcf10df36b1f1cba131b7e487ff3edafc9d6e932a3c70071b5b768bab901ff'], + }), + ('anemoi_inference', '0.6.3', { + 'modulename': 'anemoi.inference', + 'checksums': ['d957c3064560e5aef3bcf4bae6a1bd0763472eef143a1c68b88505f259d1b0c4'], + }), + ('anemoi_transform', '0.1.11', { + 'modulename': 'anemoi.transform', + 'checksums': ['79e6ad9c4bc9fffb82dec86e72da44085ab7ab7dfbb608521e8530e6f7a7e540'], + }), + ('colorcet', '3.1.0', { + 'checksums': ['2a7d59cc8d0f7938eeedd08aad3152b5319b4ba3bcb7a612398cc17a384cb296'], + }), + ('multipledispatch', '1.0.0', { + 'checksums': ['0c53cd8b077546da4e48869f49b13164bebafd0c2a5afceb6bb6a316e7fb46e4'], + }), + ('param', '2.2.1', { + 'checksums': ['e3a4ca7f3d7610615129a55dbde2e90eb67d11cef70936487b0a59717dba0bdc'], + }), + ('pyct', '0.5.0', { + 'source_tmpl': SOURCE_WHL, + 'checksums': ['a4038a8885059ab8cac6f946ea30e0b5e6bdbe0b92b6723f06737035f9d65e8c'], + }), + ('datashader', '0.18.2', { + 'checksums': ['2aa90e867a46b1e75248f32a47c5b14bb5dc869524152f88c0af8369d47359e7'], + }), + ('torchinfo', '1.8.0', { + 'checksums': ['2e911c2918603f945c26ff21a3a838d12709223dc4ccf243407bce8b6e897b46'], + }), + ('anemoi_training', '0.5.1', { + 'modulename': 'anemoi.training', + 'checksums': ['843b8c052db43bdf4bc127fbd8a542ad7d557c8a2a857fac87a1e80417b61b9e'], + }), +] + +moduleclass = 'geo'