Eager eviction even when object pinning is disabled (#8561)

* Eager eviction even when object pinning is disabled, add regression test

* Make test more robust

* lint
This commit is contained in:
Stephanie Wang
2020-06-02 11:48:03 -07:00
committed by GitHub
parent ae312af435
commit aa06c3b15a
2 changed files with 62 additions and 47 deletions
+16 -1
View File
@@ -917,12 +917,27 @@ def test_fill_object_store_exception(shutdown_only):
def test_fill_object_store_lru_fallback(shutdown_only):
ray.init(num_cpus=2, object_store_memory=10**8, lru_evict=True)
config = json.dumps({
"free_objects_batch_size": 1,
})
ray.init(
num_cpus=2,
object_store_memory=10**8,
lru_evict=True,
_internal_config=config)
@ray.remote
def expensive_task():
return np.zeros((10**8) // 2, dtype=np.uint8)
# Check that objects out of scope are cleaned up quickly.
ray.get(expensive_task.remote())
start = time.time()
for _ in range(3):
ray.get(expensive_task.remote())
end = time.time()
assert end - start < 3
oids = []
for _ in range(3):
oid = expensive_task.remote()