fix the automatic_class_in_file in windows

Signed-off-by: Benjamin Donnot <[email protected]>

Author: BDonnot

.github/workflows/main.yml

@@ -206,10 +206,10 @@ jobs:
             name: darwin,
             os: macos-latest,
           }
-          # - {
-          #   name: windows,
-          #   os: windows-2022,
-          # }
+          - {
+            name: windows,
+            os: windows-2022,
+          }
           - { 
             name: ubuntu,
             os: ubuntu-latest,

grid2op/Environment/baseEnv.py

@@ -20,8 +20,9 @@
 import warnings
 import numpy as np
 from scipy.optimize import (minimize, LinearConstraint)
-
 from abc import ABC, abstractmethod
+
+from grid2op._glop_platform_info import _IS_WINDOWS
 from grid2op.Environment._env_prev_state import _EnvPreviousState
 from grid2op.Observation import (BaseObservation,
                                  ObservationSpace,
@@ -4117,14 +4118,25 @@ def close(self):
         self._aux_close_local_dir_cls()
 
     def _aux_close_local_dir_cls(self):
-        if self._local_dir_cls is not None:
-            # I am the "keeper" of the temporary directory
-            # deleting this env should also delete the temporary directory
-            if not (hasattr(self._local_dir_cls, "_RUNNER_DO_NOT_ERASE") and not self._local_dir_cls._RUNNER_DO_NOT_ERASE):
-                # BUT if a runner uses it, then I should not delete it !
-                self._local_dir_cls.cleanup()
-                self._local_dir_cls = None
-                # In this case it's likely that the OS will clean it for grid2op with a warning...
+        if self._local_dir_cls is None:
+            return
+        # I am the "keeper" of the temporary directory
+        # deleting this env should also delete the temporary directory
+        if not (hasattr(self._local_dir_cls, "_RUNNER_DO_NOT_ERASE") and not self._local_dir_cls._RUNNER_DO_NOT_ERASE):
+            # BUT if a runner uses it, then I should not delete it !
+            self._local_dir_cls.cleanup()
+            self._local_dir_cls = None
+            # In this case it's likely that the OS will clean it for grid2op with a warning...
+        
+        # clear all imported modules (required for windows)
+        # otherwise it tries to import the classes when processes are spawned
+        # even the classes that have been "cleaned"
+        if _IS_WINDOWS and type(self)._PATH_GRID_CLASSES is not None:
+            # TODO find a better way here
+            module_name = os.path.split(type(self)._PATH_GRID_CLASSES)[-1]
+            for el in sys.modules.copy():
+                if module_name in el:
+                    del sys.modules[el]
 
     def attach_layout(self, grid_layout):
         """

grid2op/tests/automatic_classes.py

@@ -15,7 +15,6 @@
 import numpy as np
 from gymnasium.vector import AsyncVectorEnv
 
-
 import grid2op
 from grid2op._glop_platform_info import _IS_WINDOWS
 from grid2op.Runner import Runner
@@ -615,6 +614,7 @@ def test_asynch_fork(self):
 
     def test_asynch_spawn(self):
         # test I can reset everything on the same process
+        # python -m unittest grid2op.tests.automatic_classes.GymEnvAutoClassTester.test_asynch_spawn
         env1 = GymEnv(self.env)
         env2 = GymEnv(self.env)
         obs1, info1 = env1.reset()