from __future__ import annotations

__all__ = ["numpy_available", "pandas_available", "torch_available", "xarray_available"]

from pytest import mark

from coola.utils.imports import (
    is_numpy_available,
    is_pandas_available,
    is_torch_available,
    is_xarray_available,
)

numpy_available = mark.skipif(not is_numpy_available(), reason="Requires NumPy")
pandas_available = mark.skipif(not is_pandas_available(), reason="Requires pandas")
torch_available = mark.skipif(not is_torch_available(), reason="Requires PyTorch")
xarray_available = mark.skipif(not is_xarray_available(), reason="Requires xarray")
