continual-foragax 0.10.1__tar.gz → 0.10.2__tar.gz

{continual_foragax-0.10.1 → continual_foragax-0.10.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: continual-foragax
-Version: 0.10.1
+Version: 0.10.2
 Summary: A continual reinforcement learning benchmark
 Author-email: Steven Tang <stang5@ualberta.ca>
 Requires-Python: >=3.8

{continual_foragax-0.10.1 → continual_foragax-0.10.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "continual-foragax"
-version = "0.10.1"
+version = "0.10.2"
 description = "A continual reinforcement learning benchmark"
 readme = "README.md"
 authors = [
@@ -30,7 +30,7 @@ build-backend = "setuptools.build_meta"
 [tool]
 [tool.commitizen]
 name = "cz_conventional_commits"
-version = "0.10.1"
+version = "0.10.2"
 tag_format = "$version"
 version_files = ["pyproject.toml"]
 

{continual_foragax-0.10.1 → continual_foragax-0.10.2}/src/continual_foragax.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: continual-foragax
-Version: 0.10.1
+Version: 0.10.2
 Summary: A continual reinforcement learning benchmark
 Author-email: Steven Tang <stang5@ualberta.ca>
 Requires-Python: >=3.8

{continual_foragax-0.10.1 → continual_foragax-0.10.2}/src/foragax/env.py

@@ -348,7 +348,9 @@ class ForagaxEnv(environment.Environment):
 
             if is_true_mode:
                 # Apply true object borders by overlaying true colors on border pixels
-                img = apply_true_borders(img, render_grid, self.size)
+                img = apply_true_borders(
+                    img, render_grid, self.size, len(self.object_ids)
+                )
 
             # Add grid lines for world mode
             grid_color = jnp.zeros(3, dtype=jnp.uint8)
@@ -376,7 +378,9 @@ class ForagaxEnv(environment.Environment):
 
             if is_true_mode:
                 # Apply true object borders by overlaying true colors on border pixels
-                img = apply_true_borders(img, aperture, self.aperture_size)
+                img = apply_true_borders(
+                    img, aperture, self.aperture_size, len(self.object_ids)
+                )
 
             # Add grid lines for aperture mode
             grid_color = jnp.zeros(3, dtype=jnp.uint8)

{continual_foragax-0.10.1 → continual_foragax-0.10.2}/src/foragax/rendering.py

@@ -9,7 +9,10 @@ from foragax.colors import hsv_to_rgb_255
 
 
 def apply_true_borders(
-    base_img: jax.Array, true_grid: jax.Array, grid_size: Tuple[int, int]
+    base_img: jax.Array,
+    true_grid: jax.Array,
+    grid_size: Tuple[int, int],
+    num_objects: int,
 ) -> jax.Array:
     """Apply true object borders by overlaying HSV border colors on border pixels.
 
@@ -17,12 +20,12 @@ def apply_true_borders(
         base_img: Base image with object colors
         true_grid: Grid of object IDs for determining border colors
         grid_size: (height, width) of the grid
+        num_objects: Number of object types
 
     Returns:
         Image with HSV borders overlaid on border pixels
     """
     # Create HSV border colors for each object type
-    num_objects = true_grid.max() + 1  # Assume object IDs start from 0
     hues = jnp.linspace(0, 1, num_objects, endpoint=False)
 
     # Convert HSV to RGB for border colors