File size: 692 Bytes
02d27c4 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 | """Lazy torch import so `import kronos_ml` works in a pure numpy/scipy env.
Mirrors kronos_toolkit.report.plotting._mpl — torch is only touched on use."""
from __future__ import annotations
_torch = None
def torch():
global _torch
if _torch is None:
try:
import torch as _t
except Exception as e: # pragma: no cover
raise RuntimeError(
"torch is required for this surrogate; install the extra: "
"pip install 'kronos-ml[nn]'") from e
_torch = _t
return _torch
def available() -> bool:
try:
import torch # noqa: F401
return True
except Exception:
return False
|