continual-foragax 0.10.3__tar.gz → 0.11.0__tar.gz

{continual_foragax-0.10.3 → continual_foragax-0.11.0}/PKG-INFO

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

{continual_foragax-0.10.3 → continual_foragax-0.11.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "continual-foragax"
-version = "0.10.3"
+version = "0.11.0"
 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.3"
+version = "0.11.0"
 tag_format = "$version"
 version_files = ["pyproject.toml"]
 

{continual_foragax-0.10.3 → continual_foragax-0.11.0}/src/continual_foragax.egg-info/PKG-INFO

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

{continual_foragax-0.10.3 → continual_foragax-0.11.0}/src/foragax/objects.py

@@ -179,6 +179,37 @@ DEATHCAP = DefaultForagaxObject(
 )
 AGENT = DefaultForagaxObject(name="agent", blocking=True, color=(0, 0, 255))
 
+BROWN_MOREL = NormalRegenForagaxObject(
+    name="brown_morel",
+    reward=30.0,
+    collectable=True,
+    color=(63, 30, 25),
+    mean_regen_delay=300,
+    std_regen_delay=30,
+)
+BROWN_OYSTER = NormalRegenForagaxObject(
+    name="brown_oyster",
+    reward=1.0,
+    collectable=True,
+    color=(63, 30, 25),
+    mean_regen_delay=10,
+    std_regen_delay=1,
+)
+GREEN_DEATHCAP = DefaultForagaxObject(
+    name="green_deathcap",
+    reward=-1.0,
+    collectable=True,
+    color=(0, 255, 0),
+    regen_delay=(10, 10),
+)
+GREEN_FAKE = DefaultForagaxObject(
+    name="green_fake",
+    reward=0.0,
+    collectable=True,
+    color=(0, 255, 0),
+    regen_delay=(10, 10),
+)
+
 
 def create_weather_objects(
     file_index: int = 0, repeat: int = 500, multiplier: float = 1.0

{continual_foragax-0.10.3 → continual_foragax-0.11.0}/src/foragax/registry.py

@@ -10,6 +10,10 @@ from foragax.env import (
     ForagaxWorldEnv,
 )
 from foragax.objects import (
+    BROWN_MOREL,
+    BROWN_OYSTER,
+    GREEN_DEATHCAP,
+    GREEN_FAKE,
     LARGE_MOREL,
     LARGE_OYSTER,
     MEDIUM_MOREL,
@@ -28,6 +32,19 @@ ENV_CONFIGS: Dict[str, Dict[str, Any]] = {
             Biome(start=(0, 10), stop=(15, 12), object_frequencies=(0.0, 0.5)),
         ),
     },
+    "ForagaxTwoBiome-v1": {
+        "size": (15, 15),
+        "aperture_size": None,
+        "objects": (BROWN_MOREL, BROWN_OYSTER, GREEN_DEATHCAP, GREEN_FAKE),
+        "biomes": (
+            # Morel biome
+            Biome(start=(3, 0), stop=(5, 15), object_frequencies=(0.5, 0.0, 0.25, 0.0)),
+            # Oyster biome
+            Biome(
+                start=(10, 0), stop=(12, 15), object_frequencies=(0.0, 0.5, 0.0, 0.25)
+            ),
+        ),
+    },
     "ForagaxTwoBiomeSmall-v1": {
         "size": (16, 8),
         "aperture_size": None,