class Foo: def step(self, action): return {'reward': 0.5} def grader(self): return 0.5