<SYSTEM_TASK:>
Solve the following problem using Python, implementing the functions described below, one line at a time
<END_TASK>
<USER_TASK:>
Description:
 def create_time_step(cls, observation=None, done=False, raw_reward=None, processed_reward=None, action=None):
 """Creates a TimeStep with both rewards and actions as optional."""


    return cls(observation, done, raw_reward, processed_reward, action)