Fix lint errors.

tensorflow · Jan 21, 2024 · 97de036 · 97de036
1 parent 9e5f23e
commit 97de036
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 9 deletions.
diff --git a/tf_agents/environments/batched_py_environment.py b/tf_agents/environments/batched_py_environment.py
@@ -183,11 +183,7 @@ def _step(self, actions):
       return nest_utils.stack_nested_arrays(time_steps)
 
   def seed(self, seed: types.Seed) -> Any:
-    """Seeds the environment.
-
-    Args:
-      seed: Value to use as seed for the environment.
-    """
+    """Seeds the environment."""
     return self._execute(lambda env: env.seed(seed), self._envs)
 
   def get_state(self) -> Any:
@@ -197,8 +193,8 @@ def get_state(self) -> Any:
   def set_state(self, state: Sequence[Any]) -> None:
     """Restores the environment to a given `state`."""
     self._execute(
-      lambda env_and_state: env_and_state[0].set_state(env_and_state[1]),
-      zip(self._envs, state)
+        lambda env_state: env_state[0].set_state(env_state[1]),
+        zip(self._envs, state)
     )
 
   def render(self, mode="rgb_array") -> Optional[types.NestedArray]:

diff --git a/tf_agents/environments/batched_py_environment_test.py b/tf_agents/environments/batched_py_environment_test.py
@@ -130,7 +130,6 @@ def test_get_info_gym_env(self, multithreading):
   @parameterized.parameters(*COMMON_PARAMETERS)
   def test_seed_gym_env(self, multithreading):
     num_envs = 5
-    rng = np.random.RandomState()
     gym_env = self._make_batched_mock_gym_py_environment(
         multithreading, num_envs=num_envs
     )
@@ -144,7 +143,6 @@ def test_seed_gym_env(self, multithreading):
   @parameterized.parameters(*COMMON_PARAMETERS)
   def test_state_gym_env(self, multithreading):
     num_envs = 5
-    rng = np.random.RandomState()
     gym_env = self._make_batched_mock_gym_py_environment(
         multithreading, num_envs=num_envs
     )