weatherpred/zerogpu.py

"""Compatibility wrapper for HuggingFace ZeroGPU's spaces package."""

from __future__ import annotations

try:
    import spaces
except ImportError:
    class _SpacesFallback:
        @staticmethod
        def GPU(*_args, **_kwargs):
            def decorator(function):
                return function
            return decorator

    spaces = _SpacesFallback()