continual-foragax 0.20.1__tar.gz → 0.21.0__tar.gz

{continual_foragax-0.20.1 → continual_foragax-0.21.0}/PKG-INFO

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

{continual_foragax-0.20.1 → continual_foragax-0.21.0}/pyproject.toml

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

{continual_foragax-0.20.1 → continual_foragax-0.21.0}/src/continual_foragax.egg-info/PKG-INFO

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

{continual_foragax-0.20.1 → continual_foragax-0.21.0}/src/foragax/objects.py

@@ -242,6 +242,34 @@ GREEN_FAKE_2 = NormalRegenForagaxObject(
     mean_regen_delay=10,
     std_regen_delay=1,
 )
+BROWN_MOREL_UNIFORM = DefaultForagaxObject(
+    name="brown_morel",
+    reward=10.0,
+    collectable=True,
+    color=(63, 30, 25),
+    regen_delay=(90, 110),
+)
+BROWN_OYSTER_UNIFORM = DefaultForagaxObject(
+    name="brown_oyster",
+    reward=1.0,
+    collectable=True,
+    color=(63, 30, 25),
+    regen_delay=(9, 11),
+)
+GREEN_DEATHCAP_UNIFORM = DefaultForagaxObject(
+    name="green_deathcap",
+    reward=-5.0,
+    collectable=True,
+    color=(0, 255, 0),
+    regen_delay=(9, 11),
+)
+GREEN_FAKE_UNIFORM = DefaultForagaxObject(
+    name="green_fake",
+    reward=0.0,
+    collectable=True,
+    color=(0, 255, 0),
+    regen_delay=(9, 11),
+)
 
 
 def create_weather_objects(

{continual_foragax-0.20.1 → continual_foragax-0.21.0}/src/foragax/registry.py

@@ -12,12 +12,16 @@ from foragax.env import (
 from foragax.objects import (
     BROWN_MOREL,
     BROWN_MOREL_2,
+    BROWN_MOREL_UNIFORM,
     BROWN_OYSTER,
+    BROWN_OYSTER_UNIFORM,
     GREEN_DEATHCAP,
     GREEN_DEATHCAP_2,
     GREEN_DEATHCAP_3,
+    GREEN_DEATHCAP_UNIFORM,
     GREEN_FAKE,
     GREEN_FAKE_2,
+    GREEN_FAKE_UNIFORM,
     LARGE_MOREL,
     LARGE_OYSTER,
     MEDIUM_MOREL,
@@ -147,6 +151,18 @@ ENV_CONFIGS: Dict[str, Dict[str, Any]] = {
         "biomes": None,
         "nowrap": True,
     },
+    "ForagaxTwoBiome-v8": {
+        "size": None,
+        "aperture_size": None,
+        "objects": (
+            BROWN_MOREL_UNIFORM,
+            BROWN_OYSTER_UNIFORM,
+            GREEN_DEATHCAP_UNIFORM,
+            GREEN_FAKE_UNIFORM,
+        ),
+        "biomes": None,
+        "nowrap": True,
+    },
     "ForagaxTwoBiomeSmall-v1": {
         "size": (16, 8),
         "aperture_size": None,
@@ -217,7 +233,7 @@ def make(
     if nowrap is not None:
         config["nowrap"] = nowrap
 
-    if env_id == "ForagaxTwoBiome-v7":
+    if env_id in ("ForagaxTwoBiome-v7", "ForagaxTwoBiome-v8"):
         margin = aperture_size[1] // 2 + 1
         width = 2 * margin + 9
         config["size"] = (width, 15)