code: def create_time_step(cls,
                         observation=None,
                         done=False,
                         raw_reward=None,
                         processed_reward=None,
                         action=None):
      
  
      return cls(observation, done, raw_reward, processed_reward, action)
docs: Creates a TimeStep with both rewards and actions as optional.