chore: import upstream snapshot with attribution

python/ray/train/v2/api/report_config.py

@@ -0,0 +1,36 @@
+from enum import Enum
+
+from ray.util.annotations import PublicAPI
+
+
+@PublicAPI(stability="alpha")
+class CheckpointUploadMode(Enum):
+    """The manner in which we want to upload the checkpoint.
+
+    Members:
+        ASYNC: Upload checkpoint asynchronously.
+        SYNC: Upload checkpoint synchronously.
+        NO_UPLOAD: Do not upload checkpoint.
+    """
+
+    ASYNC = "ASYNC"
+    SYNC = "SYNC"
+    NO_UPLOAD = "NO_UPLOAD"
+
+    def default_delete_local_checkpoint_after_upload(self) -> bool:
+        return self == CheckpointUploadMode.ASYNC
+
+
+@PublicAPI(stability="alpha")
+class CheckpointConsistencyMode(Enum):
+    """Read semantics for checkpoint retrieval during an ongoing run.
+
+    Members:
+        COMMITTED: Block until the checkpoint from the latest ray.train.report
+            has been uploaded and committed.
+        VALIDATED: Block until the checkpoint from the latest ray.train.report
+            has been uploaded and validated.
+    """
+
+    COMMITTED = "COMMITTED"
+    VALIDATED = "VALIDATED"