code: def test_pickling_keeps_goal(env_type):
      env = env_type()
      task = env.sample_tasks(1)[0]
      env.set_task(task)
      env_clone = pickle.loads(pickle.dumps(env))
      assert env._task == env_clone._task
  
imports: import pickle
  import numpy
  import pytest
  from garage.envs.mujoco.half_cheetah_dir_env import HalfCheetahDirEnv
  from garage.envs.mujoco.half_cheetah_vel_env import HalfCheetahVelEnv
label_desc: Use image node_id 2 for calling a global function with example usage: test_pickling_keeps_goal(env_type) without return types