Increase WFCEnv max_attempt limit

Minor cleanup
jysdoran · Aug 16, 2023 · f1bb1e3 · f1bb1e3
1 parent 456d948
commit f1bb1e3
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 3 deletions.
diff --git a/minigrid/envs/wfc/graphtransforms.py b/minigrid/envs/wfc/graphtransforms.py
@@ -15,7 +15,6 @@
 class EdgeDescriptor:
     between: tuple[str, str] | tuple[str]
     structure: str | None = None
-    weight = None
 
 
 # This is maybe general enough to be in utils

diff --git a/minigrid/envs/wfc/wfcenv.py b/minigrid/envs/wfc/wfcenv.py
@@ -102,7 +102,7 @@ def __init__(
         if size < 3:
             raise ValueError(f"Grid size must be at least 3 (currently {size})")
         self.size = size
-        self.max_attempts = 100
+        self.max_attempts = 1000
 
         if max_steps is None:
             max_steps = self.size * 20

diff --git a/tests/test_wfc/conftest.py b/tests/test_wfc/conftest.py
@@ -22,7 +22,7 @@ def img_redmaze(resources: Resources) -> array:
         import imageio  # type: ignore
 
         pattern = resources.get_pattern("RedMaze.png")
-        img = imageio.imread(pattern)
+        img = imageio.v2.imread(pattern)
     except ImportError:
         b = [0, 0, 0]
         w = [255, 255, 255]