[Object Spilling] Initial Iteration of S3 adapter. (#11379)

* Finished the first iteration.

* Removed unnecessary code.

* Smartopen impl.

* Make sure tests passed.

* Addressed code review.

* Addressed code review.

* Fix issues.

* Fix issues.
This commit is contained in:
SangBin Cho
2020-10-30 14:47:07 -07:00
committed by GitHub
parent 7aade469d0
commit 71c5089854
4 changed files with 229 additions and 44 deletions
+8
View File
@@ -320,3 +320,11 @@ class RayParams:
if numpy_major <= 1 and numpy_minor < 16:
logger.warning("Using ray with numpy < 1.16.0 will result in slow "
"serialization. Upgrade numpy if using with ray.")
# Make sure object spilling configuration is applicable.
object_spilling_config = self.object_spilling_config or {}
if object_spilling_config:
from ray import external_storage
# Validate external storage usage.
external_storage.setup_external_storage(object_spilling_config)
external_storage.reset_external_storage()