continual-foragax 0.42.2__py3-none-any.whl → 0.43.1__py3-none-any.whl

{continual_foragax-0.42.2.dist-info → continual_foragax-0.43.1.dist-info}/METADATA

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

{continual_foragax-0.42.2.dist-info → continual_foragax-0.43.1.dist-info}/RECORD

@@ -2,7 +2,7 @@ foragax/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 foragax/colors.py,sha256=rqNPiywP4Nvr0POhsGpasRk-nMMTS3DOwFRUgperlUk,2065
 foragax/env.py,sha256=XRrdQew23Vl3kcm0U68da7736J4Xgz9y0WxcrqKflHk,69788
 foragax/objects.py,sha256=fVUPJYT7alvUl6KGZHZPYAkMVwup8s2BV6lWY5o7CdQ,27198
-foragax/registry.py,sha256=3IA0BKHLviWE_wyeiNREA4c0LbIXc-sUOCTo-ZZB9mQ,21173
+foragax/registry.py,sha256=Ce1Fkm1WU-4CpIWqLKMNcgMa0I-AS9a1vkVjeLOFnTM,24034
 foragax/rendering.py,sha256=E55NxxbKyIaIo77XJzXHqd1u3gLrV-WTPwjqCawNgMs,5776
 foragax/weather.py,sha256=KNAiwuFz8V__6G75vZIWQKPocLzXqxXn-Vt4TbHIpcA,1258
 foragax/data/ECA_non-blended_custom/TG_SOUID100897.txt,sha256=N7URbX6VlCZvCboUogYjMzy1I-0cfNPOn0QTLSHHfQ0,1776751
@@ -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.42.2.dist-info/METADATA,sha256=UUTUC9Nd87IDO0fDAFj2KEfytvuy9lTKC4c3aAY4ZOU,4713
-continual_foragax-0.42.2.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
-continual_foragax-0.42.2.dist-info/entry_points.txt,sha256=Qiu6iE_XudrDO_bVAMeA435h4PO9ourt8huvSHiuMPc,41
-continual_foragax-0.42.2.dist-info/top_level.txt,sha256=-z3SDK6RfLIcLI24n8rdbeFzlVY3hunChzlu-v1Fncs,8
-continual_foragax-0.42.2.dist-info/RECORD,,
+continual_foragax-0.43.1.dist-info/METADATA,sha256=JS_3hFZHPdxLGXJtxwGGXEnlpCGeVaacY8avWycyvos,4713
+continual_foragax-0.43.1.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+continual_foragax-0.43.1.dist-info/entry_points.txt,sha256=Qiu6iE_XudrDO_bVAMeA435h4PO9ourt8huvSHiuMPc,41
+continual_foragax-0.43.1.dist-info/top_level.txt,sha256=-z3SDK6RfLIcLI24n8rdbeFzlVY3hunChzlu-v1Fncs,8
+continual_foragax-0.43.1.dist-info/RECORD,,

foragax/registry.py

@@ -31,11 +31,15 @@ from foragax.objects import (
     LARGE_MOREL,
     LARGE_OYSTER,
     MEDIUM_MOREL,
+    WALL,
     create_fourier_objects,
     create_sine_biome_objects,
     create_weather_objects,
 )
 
+BIG_WIDTH = 15
+BIG_OFFSET = BIG_WIDTH // 2 + 1
+
 ENV_CONFIGS: Dict[str, Dict[str, Any]] = {
     "ForagaxWeather-v1": {
         "size": (15, 15),
@@ -165,6 +169,70 @@ ENV_CONFIGS: Dict[str, Dict[str, Any]] = {
         "biome_consumption_threshold": 10000,
         "dynamic_biome_spawn_empty": 0.4,
     },
+    "ForagaxBig-v1": {
+        "size": (4 * BIG_WIDTH, 4 * BIG_WIDTH),
+        "aperture_size": None,
+        "objects": None,
+        "biomes": (
+            Biome(
+                start=(BIG_OFFSET, BIG_OFFSET),
+                stop=(BIG_OFFSET + BIG_WIDTH, BIG_OFFSET + BIG_WIDTH),
+                object_frequencies=(0.4, 0.0),
+            ),
+            Biome(
+                start=(BIG_OFFSET, BIG_OFFSET + 2 * BIG_WIDTH),
+                stop=(BIG_OFFSET + BIG_WIDTH, BIG_OFFSET + 3 * BIG_WIDTH),
+                object_frequencies=(0.4, 0.0),
+            ),
+            Biome(
+                start=(BIG_OFFSET + 2 * BIG_WIDTH, BIG_OFFSET),
+                stop=(BIG_OFFSET + 3 * BIG_WIDTH, BIG_OFFSET + BIG_WIDTH),
+                object_frequencies=(0.4, 0.0),
+            ),
+            Biome(
+                start=(BIG_OFFSET + 2 * BIG_WIDTH, BIG_OFFSET + 2 * BIG_WIDTH),
+                stop=(BIG_OFFSET + 3 * BIG_WIDTH, BIG_OFFSET + 3 * BIG_WIDTH),
+                object_frequencies=(0.4, 0.0),
+            ),
+            Biome(
+                start=(BIG_OFFSET + BIG_WIDTH // 2, BIG_OFFSET + BIG_WIDTH // 2),
+                stop=(BIG_OFFSET + BIG_WIDTH // 2 + 1, BIG_OFFSET + BIG_WIDTH // 2 + 1),
+                object_frequencies=(0.0, 1.0),
+            ),
+            Biome(
+                start=(BIG_OFFSET + BIG_WIDTH // 2, BIG_OFFSET + 5 * BIG_WIDTH // 2),
+                stop=(
+                    BIG_OFFSET + BIG_WIDTH // 2 + 1,
+                    BIG_OFFSET + 5 * BIG_WIDTH // 2 + 1,
+                ),
+                object_frequencies=(0.0, 1.0),
+            ),
+            Biome(
+                start=(BIG_OFFSET + 5 * BIG_WIDTH // 2, BIG_OFFSET + BIG_WIDTH // 2),
+                stop=(
+                    BIG_OFFSET + 5 * BIG_WIDTH // 2 + 1,
+                    BIG_OFFSET + BIG_WIDTH // 2 + 1,
+                ),
+                object_frequencies=(0.0, 1.0),
+            ),
+            Biome(
+                start=(
+                    BIG_OFFSET + 5 * BIG_WIDTH // 2,
+                    BIG_OFFSET + 5 * BIG_WIDTH // 2,
+                ),
+                stop=(
+                    BIG_OFFSET + 5 * BIG_WIDTH // 2 + 1,
+                    BIG_OFFSET + 5 * BIG_WIDTH // 2 + 1,
+                ),
+                object_frequencies=(0.0, 1.0),
+            ),
+        ),
+        "nowrap": False,
+        "deterministic_spawn": True,
+        "dynamic_biomes": True,
+        "biome_consumption_threshold": 10000,
+        "dynamic_biome_spawn_empty": 1.0,
+    },
     "ForagaxTwoBiome-v1": {
         "size": (15, 15),
         "aperture_size": None,
@@ -633,6 +701,16 @@ def make(
             regen_delay=(9, 11),
             reward_repeat=1000,
         )[:1]
+    if env_id == "ForagaxBig-v1":
+        config["objects"] = (
+            create_fourier_objects(
+                num_fourier_terms=10,
+                reward_delay=reward_delay,
+                regen_delay=(9, 11),
+                reward_repeat=1000,
+            )[0],
+            WALL,
+        )
 
     if env_id == "ForagaxSineTwoBiome-v1":
         biome1_oyster, biome1_deathcap, biome2_oyster, biome2_deathcap = (