in lm_human_preferences/utils/core.py [0:0]
def from_dict(stats):
for k, v in stats.items():
if v.dtype != tf.float32:
raise ValueError('Statistic %s has dtype %r, expected %r' % (k, v.dtype, tf.float32))
keys = tuple(sorted(stats.keys()))
flat = tf.stack([stats[k] for k in keys])
return FlatStats(keys, flat)