continual-foragax 0.42.2__tar.gz → 0.43.1__tar.gz

{continual_foragax-0.42.2 → continual_foragax-0.43.1}/PKG-INFO

@@ -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 → continual_foragax-0.43.1}/pyproject.toml

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

{continual_foragax-0.42.2 → continual_foragax-0.43.1}/src/continual_foragax.egg-info/PKG-INFO

@@ -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 → continual_foragax-0.43.1}/src/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 = (