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

[tune] Remove unused clean_trial_placement_group. (#19960)

Browse source
This commit is contained in:
xwjiang2010 2021-11-04 08:55:42 -07:00 • committed by GitHub
parent bcb63961d9
commit f1179cbccd
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 0 additions and 28 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

28
python/ray/tune/utils/placement_groups.py
View file

@ -648,34 +648,6 @@ class PlacementGroupManager:
return trial_pg return trial_pg
def clean_trial_placement_group(
self, trial: "Trial") -> Optional[PlacementGroup]:
"""Remove reference to placement groups associated with a trial.
Returns an associated placement group. If the trial was scheduled, this
is the placement group it was scheduled on. If the trial was not
scheduled, it will first try to return a staging placement group. If
there is no staging placement group, it will return a ready placement
group that is not yet being used by another trial.
Args:
trial ("Trial"): :obj:`Trial` object.
Returns:
PlacementGroup or None.
"""
pgf = trial.placement_group_factory
if trial in self._in_use_trials:
# "Trial" was in use. Just return its placement group.
trial_pg = self._in_use_trials.pop(trial)
self._in_use_pgs.pop(trial_pg)
else:
trial_pg = self._unstage_unused_pg(pgf)
return trial_pg
def in_staging_grace_period(self): def in_staging_grace_period(self):
return self._staging_futures and self._grace_period and time.time( return self._staging_futures and self._grace_period and time.time(
) <= self._latest_staging_start_time + self._grace_period ) <= self._latest_staging_start_time + self._grace_period