ocf-data-sampler>=0.5.20
numpy
pandas
matplotlib
xarray
h5netcdf
torch>=2.0.0
lightning
torchvision
typer
sqlalchemy
fsspec[s3]
wandb
huggingface-hub
tqdm
omegaconf
hydra-core
rich
einops
safetensors
