continual-foragax 0.10.1__py3-none-any.whl → 0.10.2__py3-none-any.whl

{continual_foragax-0.10.1.dist-info → continual_foragax-0.10.2.dist-info}/METADATA

@@ -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.dist-info → continual_foragax-0.10.2.dist-info}/RECORD

@@ -1,9 +1,9 @@
 foragax/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 foragax/colors.py,sha256=rqNPiywP4Nvr0POhsGpasRk-nMMTS3DOwFRUgperlUk,2065
-foragax/env.py,sha256=X9oc60xNL4uTFnbt_BynN_c3XmVa9MYbSclW-g4qQoc,18628
+foragax/env.py,sha256=Oo0jlvDXZTzV92yKJ08krRRg5yD1wSsvuIvxUrzMxu4,18748
 foragax/objects.py,sha256=CyBxrykTxpHCI_2hE9jE8mG4TU8R7VxzKdQ5mtxkEqU,6004
 foragax/registry.py,sha256=7_RDXvm_3RNO7culBLGkE0jH8Wk_q6jbMv72dZx4JO8,2722
-foragax/rendering.py,sha256=KAoQpdndy5JDQlyG0c5QDHuH-_Tfy5RuVlDtndnHVjc,2765
+foragax/rendering.py,sha256=bms7wvBZTofoR-K-2QD2Ggeed7Viw8uwAEiEpEM3eSo,2768
 foragax/weather.py,sha256=KNAiwuFz8V__6G75vZIWQKPocLzXqxXn-Vt4TbHIpcA,1258
 foragax/data/ECA_non-blended_custom/TG_SOUID100897.txt,sha256=N7URbX6VlCZvCboUogYjMzy1I-0cfNPOn0QTLSHHfQ0,1776751
 foragax/data/ECA_non-blended_custom/TG_SOUID100928.txt,sha256=AaJMWisVu2YPlZFwvetzHZBI2DqWqUYmp8BVzx6gWZI,817991
@@ -128,8 +128,8 @@ foragax/data/ECA_non-blended_custom/TG_SOUID156887.txt,sha256=juzTPgJoJxfqmZkorL
 foragax/data/ECA_non-blended_custom/elements.txt,sha256=OtcUBoDAHxuln79BPKGu0tsQxG_5G2BfAX3Ck130kEA,4507
 foragax/data/ECA_non-blended_custom/metadata.txt,sha256=nudnmOCy5cPJfSXt_IjyX0S5-T7NkCZREICZSimqeqc,48260
 foragax/data/ECA_non-blended_custom/sources.txt,sha256=1j3lSmINAoCMqPqFrHfZJriOz6sTYZNOhXzUwvTLas0,20857
-continual_foragax-0.10.1.dist-info/METADATA,sha256=tgSSa5FtfA6qRBCZ_L3r3XnsfDpjhKMTRFJ9TL_OfCo,4897
-continual_foragax-0.10.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-continual_foragax-0.10.1.dist-info/entry_points.txt,sha256=Qiu6iE_XudrDO_bVAMeA435h4PO9ourt8huvSHiuMPc,41
-continual_foragax-0.10.1.dist-info/top_level.txt,sha256=-z3SDK6RfLIcLI24n8rdbeFzlVY3hunChzlu-v1Fncs,8
-continual_foragax-0.10.1.dist-info/RECORD,,
+continual_foragax-0.10.2.dist-info/METADATA,sha256=fpB7XhrzvckiVovAFHYB3GUWdRKr2DhiUTRCLS815A4,4897
+continual_foragax-0.10.2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+continual_foragax-0.10.2.dist-info/entry_points.txt,sha256=Qiu6iE_XudrDO_bVAMeA435h4PO9ourt8huvSHiuMPc,41
+continual_foragax-0.10.2.dist-info/top_level.txt,sha256=-z3SDK6RfLIcLI24n8rdbeFzlVY3hunChzlu-v1Fncs,8
+continual_foragax-0.10.2.dist-info/RECORD,,

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)

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