Traceback (most recent call last):
  File "/home/frank14f/Frank_LBM/scripts/d1a3o12_imit.py", line 55, in <module>
    test_obs, test_rewards, test_dones, info = test_env.step(test_action)
  File "/home/frank14f/anaconda3/envs/pycuda_3_10/lib/python3.10/site-packages/stable_baselines3/common/vec_env/base_vec_env.py", line 206, in step
    return self.step_wait()
  File "/home/frank14f/anaconda3/envs/pycuda_3_10/lib/python3.10/site-packages/stable_baselines3/common/vec_env/dummy_vec_env.py", line 58, in step_wait
    obs, self.buf_rews[env_idx], terminated, truncated, self.buf_infos[env_idx] = self.envs[env_idx].step(
  File "/home/frank14f/anaconda3/envs/pycuda_3_10/lib/python3.10/site-packages/stable_baselines3/common/monitor.py", line 94, in step
    observation, reward, terminated, truncated, info = self.env.step(action)
  File "/home/frank14f/Frank_LBM/scripts/gym_env_imit.py", line 190, in step
    run_flow_field(action)
  File "/home/frank14f/Frank_LBM/scripts/gym_env_imit.py", line 117, in run_flow_field
    self.flow_field.run(SAMPLE_INTERVAL, temp)
  File "/home/frank14f/Frank_LBM/CelerisLab/driver.py", line 254, in run
    cuda.memcpy_htod_async(self.action_gpu, action_pinned, stream)
KeyboardInterrupt