__init__.py 588 B

1234567891011121314151617181920
  1. from joblib.parallel import register_parallel_backend
  2. def register_ray():
  3. """Register Ray Backend to be called with parallel_backend("ray")."""
  4. try:
  5. from ray.util.joblib.ray_backend import RayBackend
  6. register_parallel_backend("ray", RayBackend)
  7. except ImportError:
  8. msg = (
  9. "To use the ray backend you must install ray."
  10. "Try running 'pip install ray'."
  11. "See https://docs.ray.io/en/master/installation.html"
  12. "for more information."
  13. )
  14. raise ImportError(msg)
  15. __all__ = ["register_ray"]