Fix some small bugs in gymnasium.vector (#1112)

Co-authored-by: leoxhwang <[email protected]>

Author: xihuai18

gymnasium/vector/async_vector_env.py

@@ -650,7 +650,7 @@ def _async_worker(
     env_fn: callable,
     pipe: Connection,
     parent_pipe: Connection,
-    shared_memory: bool,
+    shared_memory: multiprocessing.Array | dict[str, Any] | tuple[Any, ...],
     error_queue: Queue,
 ):
     env = env_fn()
@@ -699,7 +699,7 @@ def _async_worker(
                 break
             elif command == "_call":
                 name, args, kwargs = data
-                if name in ["reset", "step", "close", "set_wrapper_attr"]:
+                if name in ["reset", "step", "close", "_setattr", "_check_spaces"]:
                     raise ValueError(
                         f"Trying to call function `{name}` with `call`, use `{name}` directly instead."
                     )

gymnasium/vector/sync_vector_env.py

@@ -152,6 +152,8 @@ def reset(
             self.single_observation_space, observations, self._observations
         )
 
+        self._autoreset_envs = np.zeros((self.num_envs,), dtype=np.bool_)
+
         return deepcopy(self._observations) if self.copy else self._observations, infos
 
     def step(