ocf_datapipes>=3.3.6
ocf_ml_metrics>=0.0.11
numpy
pandas
matplotlib
xarray
ipykernel
h5netcdf
torch>=2.0.0
lightning>=2.0.1
torchvision
pytest
pytest-cov
typer
sqlalchemy
fsspec[s3]
wandb
huggingface-hub
tensorboard
tqdm
omegaconf
hydra-core
python-dotenv
hydra-optuna-sweeper
rich
gcsfs

[all_models]
pytorch-tabnet
efficientnet_pytorch
