diff --git a/.pylintrc b/.pylintrc
index 61e2990..4d07a77 100644
--- 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
index aa7acbb..58b3809 100755
--- 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)