Skip to content
Permalink
Browse files

Return 0 rather than NaN for teammate stats if there is no teammate.

PiperOrigin-RevId: 248052410
  • Loading branch information...
liusiqi43 authored and alimuldal committed May 14, 2019
1 parent 37a17bf commit e681949dda5ef1b05acf4fad99a134101469d8e7
Showing with 1 addition and 1 deletion.
  1. +1 −1 dm_control/locomotion/soccer/observables.py
@@ -299,7 +299,7 @@ def _stats_avg_teammate_dist(physics):
physics.bind(player.walker.root_body).xpos -
physics.bind(other.walker.root_body).xpos)
teammate_dists.append(dist)
return np.mean(teammate_dists) if teammate_dists else np.nan
return np.mean(teammate_dists) if teammate_dists else 0.

player.walker.observables.add_observable(
'stats_home_avg_teammate_dist',

0 comments on commit e681949

Please sign in to comment.
You can’t perform that action at this time.