env = gym.make('MountainCar-v0', render_mode=RENDER_MODE)  
state_dim = env.observation_space.shape[0]
action_dim = env.action_space.n