wassname/ray
1
0
Fork 0
mirror of https://github.com/wassname/ray.git synced 2026-07-05 20:32:50 +08:00
Code Issues Packages Projects Releases Wiki Activity

[RLLib] WindowStat bug fix (#9213)

Browse Source 
* WindowStat error catching, which processes NaNs properly instead of erroring. This ought to resolve issue #7910.
https://github.com/ray-project/ray/issues/7910
This commit is contained in:
Tanay Wakhare
2020-07-12 17:01:32 -04:00
committed by GitHub
parent 3536d8e4b3
commit 15aa08a3d1
1 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
rllib/utils/window_stat.py
+6 -6
View File
@@ -16,13 +16,13 @@ class WindowStat:
def stats(self):
if not self.count:
quantiles = []
_quantiles = []
else:
quantiles = np.percentile(self.items[:self.count],
[0, 10, 50, 90, 100]).tolist()
_quantiles = np.nanpercentile(self.items[:self.count],
[0, 10, 50, 90, 100]).tolist()
return {
self.name + "_count": int(self.count),
self.name + "_mean": float(np.mean(self.items[:self.count])),
self.name + "_std": float(np.std(self.items[:self.count])),
self.name + "_quantiles": quantiles,
self.name + "_mean": float(np.nanmean(self.items[:self.count])),
self.name + "_std": float(np.nanstd(self.items[:self.count])),
self.name + "_quantiles": _quantiles,
}