We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 2afb6f8 commit 37577eaCopy full SHA for 37577ea
compiler_opt/rl/env.py
@@ -179,7 +179,7 @@ def _get_step_type() -> StepType:
179
tv_dict = {}
180
for fv in obs.feature_values:
181
array = fv.to_numpy()
182
- tv_dict[fv.spec.name] = np.reshape(array, newshape=fv.spec.shape)
+ tv_dict[fv.spec.name] = np.reshape(array, fv.spec.shape)
183
return TimeStep(
184
obs=tv_dict,
185
reward={obs.context: obs.score} if obs.score else None,
0 commit comments