hiro/hopsflow
1
0
Fork 0
mirror of https://github.com/vale981/hopsflow synced 2025-03-05 16:51:39 -05:00
Code Issues Projects Releases Packages Wiki Activity Actions

unconditional gc sleep

Browse source
This commit is contained in:
Valentin Boettcher 2022-06-16 11:53:15 +02:00
parent e32a367a7a
commit b7940c1d80
1 changed files with 3 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
hopsflow/util.py
View file

@ -648,6 +648,7 @@ def ensemble_mean(
overwrite_cache: bool = False, overwrite_cache: bool = False,
chunk_size: Optional[int] = None, chunk_size: Optional[int] = None,
in_flight: Optional[int] = None, in_flight: Optional[int] = None,
gc_sleep: float = 0.1,
) -> EnsembleValue: ) -> EnsembleValue:
results = [] results = []
first_result = function(next(arg_iter)) first_result = function(next(arg_iter))
@ -776,9 +777,8 @@ def ensemble_mean(
highest_index += 1 highest_index += 1
amount = gc.collect() gc.collect()
if amount > 0: time.sleep(gc_sleep)
time.sleep(0.1)
if next_val: if next_val:
chunk_ref = ray.put(next_val[0]) chunk_ref = ray.put(next_val[0])