Fix pylint errors

MatPoliquin · Jan 30, 2025 · ad09ab7 · ad09ab7
1 parent 74c448e
commit ad09ab7
Show file tree

Hide file tree

Showing 4 changed files with 7 additions and 17 deletions.
diff --git a/custom_trainers/nhl941on1_trainer.py b/custom_trainers/nhl941on1_trainer.py
@@ -1,11 +1,6 @@
 """
 Game specfic training script for NHL94 1 on 1 on Genesis
 """
-
-#hack to disable flood of warnings
-import warnings
-warnings.filterwarnings("ignore")
-
 import os
 import sys
 import datetime

diff --git a/custom_trainers/wwf_trainer.py b/custom_trainers/wwf_trainer.py
@@ -2,10 +2,6 @@
 Game specfic training script for WWF Wrestlemania The Arcade Game on Genesis
 """
 
-#hack to disable flood of warnings
-import warnings
-warnings.filterwarnings("ignore")
-
 import os
 import sys
 import datetime

diff --git a/game_wrappers/compare_model_display.py b/game_wrappers/compare_model_display.py
@@ -3,17 +3,17 @@
 """
 
 import os
+from os import environ
 import datetime
 import math
 import argparse
-from os import environ
-environ['PYGAME_HIDE_SUPPORT_PROMPT'] = '1'
-import pygame
-import pygame.freetype
 import cv2
 import gymnasium as gym
 import numpy as np
 import retro
+environ['PYGAME_HIDE_SUPPORT_PROMPT'] = '1'
+import pygame # pylint: disable=wrong-import-position
+import pygame.freetype # pylint: disable=wrong-import-position
 
 FB_WIDTH = 1920
 FB_HEIGHT = 1080

diff --git a/game_wrappers/nhl94_display.py b/game_wrappers/nhl94_display.py
@@ -5,18 +5,17 @@
 import os, datetime
 import argparse
 from os import environ
-environ['PYGAME_HIDE_SUPPORT_PROMPT'] = '1'
 import math
 import sys
-import pygame
-import pygame.freetype
 import cv2
 import retro
 import gymnasium as gym
 import numpy as np
 from game_wrappers.nhl94_rf import rf_defensezone, rf_scoregoal
 from game_wrappers.nhl94_gamestate import NHL94GameState
-
+environ['PYGAME_HIDE_SUPPORT_PROMPT'] = '1'
+import pygame # pylint: disable=wrong-import-position
+import pygame.freetype # pylint: disable=wrong-import-position
 
 class NHL94GameDisplayEnv(gym.Wrapper):
     def __init__(self, env, args, total_params, nn_type, button_names):