python – PY_environment ‘time_step’ doesn’t match ‘time_step_spec’ – I can’t spot the difference

I’m trying to create a custom tf-agents environment for trading. When I try to validate it by calling utils.validate_py_environment(environment, episodes=1), I’m getting a ValueError ‘time_step’ doesn’t match ‘time_step_spec’ . I’ve been trying to spot the difference for a while now but I can’t seem to find it. Am I missing something? My Environment class TradingEnv(py_environment.PyEnvironment): … Read more