Merge pull request #17 from UoA-CARES/dev/pylint-update

numpy/torch to pylintrc
UoA-CARES · Dec 7, 2023 · 7edb8b1 · 7edb8b1
2 parents e7e0f77 + 16e6926
commit 7edb8b1
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/.pylintrc b/.pylintrc
@@ -16,4 +16,4 @@ max-line-length=130
 extension-pkg-whitelist=cv2
 
 [TYPECHECK]
-generated-members=cv2.*
+generated-members=cv2.*, numpy.*, torch.*
diff --git a/scripts/train.py b/scripts/train.py
@@ -46,9 +46,8 @@ def main():
         logging.info(
             f"Training iteration {training_iteration+1}/{len(training_config.seeds)} with Seed: {seed}"
         )
-        env = env_factory.create_environment(
-            env_config
-        )  # This line should be here for seed consistency issues
+        # This line should be here for seed consistency issues
+        env = env_factory.create_environment(env_config)
         hlp.set_seed(seed)
         env.set_seed(seed)