__init__.py 533 B

123456789101112131415161718192021
  1. from ray.air.config import (
  2. CheckpointConfig,
  3. FailureConfig,
  4. RunConfig,
  5. ScalingConfig,
  6. )
  7. from ray.air.data_batch_type import DataBatchType
  8. from ray.air.execution.resources.request import AcquiredResources, ResourceRequest
  9. from ray.air.result import Result
  10. import ray.data # noqa: F401 # TODO: This is a hack to avoid circular import
  11. __all__ = [
  12. "DataBatchType",
  13. "RunConfig",
  14. "Result",
  15. "ScalingConfig",
  16. "FailureConfig",
  17. "CheckpointConfig",
  18. "AcquiredResources",
  19. "ResourceRequest",
  20. ]