Commit e0df9928 authored by Avisek Naug's avatar Avisek Naug 🎨
Browse files

reward returned as part of info dict

parent 0b28c525
......@@ -116,6 +116,7 @@ class Env(gym.Env):
step_info = {'rl_energy': rlenergy,
'old_energy': oldenergy,
'oat': self.state[0],
'dat': controlact}
self.counter += 1
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment