Formatting

Author: Rocamonde

ci/check_typeignore.py

@@ -13,8 +13,6 @@
 import sys
 from typing import List
 
-import click
-
 # Regex to match a "# type: ignore" comment not followed by a reason.
 TYPE_IGNORE_COMMENT = re.compile(r"#\s*type:\s*ignore\s*(?![^\[]*\[)")
 
@@ -75,6 +73,7 @@ def parse_args():
     args = parser.parse_args()
     return parser, args
 
+
 def main():
     """Check for invalid "# type: ignore" comments."""
     parser, args = parse_args()

ci/clean_notebooks.py

@@ -90,7 +90,7 @@ def parse_args():
 
 
 def get_files(input_paths: List):
-    """build list of files to scan from list of paths and files"""
+    """Build list of files to scan from list of paths and files."""
     files = []
     for file in input_paths:
         if file.is_dir():

src/imitation/policies/replay_buffer_wrapper.py

@@ -4,7 +4,7 @@
 
 import numpy as np
 from gym import spaces
-from stable_baselines3.common.buffers import BaseBuffer, ReplayBuffer
+from stable_baselines3.common.buffers import ReplayBuffer
 from stable_baselines3.common.type_aliases import ReplayBufferSamples
 
 from imitation.rewards.reward_function import RewardFn

src/imitation/util/util.py

@@ -177,9 +177,8 @@ def make_seeds(
     Returns:
         A list of n random seeds.
     """
-    seeds: List[int] = rng.integers(
-        0, (1 << 31) - 1, (n if n is not None else 1,)
-    ).tolist()
+    seeds_arr = rng.integers(0, (1 << 31) - 1, (n if n is not None else 1,))
+    seeds: List[int] = seeds_arr.tolist()
     if n is None:
         return seeds[0]
     else: