continual-foragax 0.14.0__py3-none-any.whl → 0.16.0__py3-none-any.whl

{continual_foragax-0.14.0.dist-info → continual_foragax-0.16.0.dist-info}/METADATA

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

{continual_foragax-0.14.0.dist-info → continual_foragax-0.16.0.dist-info}/RECORD

@@ -1,8 +1,8 @@
 foragax/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 foragax/colors.py,sha256=rqNPiywP4Nvr0POhsGpasRk-nMMTS3DOwFRUgperlUk,2065
 foragax/env.py,sha256=EyT6KY0d0mXNh6yw10V-8SJVAdyPAGKtRdFV4wXq-JM,19836
-foragax/objects.py,sha256=-XkuZVbbSTFNJ02f6HsH4GbxZ6YrUetiY39r9Zw4UWE,6967
-foragax/registry.py,sha256=-Vj8fXlW8xbpCM6ZhEp0hc3EIGBUqP7tIaI2dwWv7RE,4972
+foragax/objects.py,sha256=wyTqqmfksvGXdxw8rxsUn06CrB22hPIDdoQTSMI1qd8,7657
+foragax/registry.py,sha256=iAM6lBwUB4h-3jVz8raZcSCmiBXwsptyc7TJbgxprAo,6033
 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
@@ -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.14.0.dist-info/METADATA,sha256=8IVQW9CR42G8PrAfd5oi4hCLLAULf5JUSNaAkYNS14k,4897
-continual_foragax-0.14.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-continual_foragax-0.14.0.dist-info/entry_points.txt,sha256=Qiu6iE_XudrDO_bVAMeA435h4PO9ourt8huvSHiuMPc,41
-continual_foragax-0.14.0.dist-info/top_level.txt,sha256=-z3SDK6RfLIcLI24n8rdbeFzlVY3hunChzlu-v1Fncs,8
-continual_foragax-0.14.0.dist-info/RECORD,,
+continual_foragax-0.16.0.dist-info/METADATA,sha256=spiV8s6tTWMtx0CWxpIecNNbmmYF16t0_hHg2gEowwo,4897
+continual_foragax-0.16.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+continual_foragax-0.16.0.dist-info/entry_points.txt,sha256=Qiu6iE_XudrDO_bVAMeA435h4PO9ourt8huvSHiuMPc,41
+continual_foragax-0.16.0.dist-info/top_level.txt,sha256=-z3SDK6RfLIcLI24n8rdbeFzlVY3hunChzlu-v1Fncs,8
+continual_foragax-0.16.0.dist-info/RECORD,,

foragax/objects.py

@@ -189,6 +189,14 @@ BROWN_MOREL = NormalRegenForagaxObject(
     mean_regen_delay=300,
     std_regen_delay=30,
 )
+BROWN_MOREL_2 = NormalRegenForagaxObject(
+    name="brown_morel",
+    reward=10.0,
+    collectable=True,
+    color=(63, 30, 25),
+    mean_regen_delay=100,
+    std_regen_delay=10,
+)
 BROWN_OYSTER = NormalRegenForagaxObject(
     name="brown_oyster",
     reward=1.0,
@@ -204,6 +212,21 @@ GREEN_DEATHCAP = DefaultForagaxObject(
     color=(0, 255, 0),
     regen_delay=(10, 10),
 )
+GREEN_DEATHCAP_2 = DefaultForagaxObject(
+    name="green_deathcap",
+    reward=-5.0,
+    collectable=True,
+    color=(0, 255, 0),
+    regen_delay=(10, 10),
+)
+GREEN_DEATHCAP_3 = DefaultForagaxObject(
+    name="green_deathcap",
+    reward=-5.0,
+    collectable=True,
+    color=(0, 255, 0),
+    mean_regen_delay=10,
+    std_regen_delay=1,
+)
 GREEN_FAKE = DefaultForagaxObject(
     name="green_fake",
     reward=0.0,
@@ -211,6 +234,14 @@ GREEN_FAKE = DefaultForagaxObject(
     color=(0, 255, 0),
     regen_delay=(10, 10),
 )
+GREEN_FAKE_2 = DefaultForagaxObject(
+    name="green_fake",
+    reward=0.0,
+    collectable=True,
+    color=(0, 255, 0),
+    mean_regen_delay=10,
+    std_regen_delay=1,
+)
 
 
 def create_weather_objects(

foragax/registry.py

@@ -11,9 +11,13 @@ from foragax.env import (
 )
 from foragax.objects import (
     BROWN_MOREL,
+    BROWN_MOREL_2,
     BROWN_OYSTER,
     GREEN_DEATHCAP,
+    GREEN_DEATHCAP_2,
+    GREEN_DEATHCAP_3,
     GREEN_FAKE,
+    GREEN_FAKE_2,
     LARGE_MOREL,
     LARGE_OYSTER,
     MEDIUM_MOREL,
@@ -73,6 +77,34 @@ ENV_CONFIGS: Dict[str, Dict[str, Any]] = {
         ),
         "nowrap": True,
     },
+    "ForagaxTwoBiome-v3": {
+        "size": (15, 15),
+        "aperture_size": None,
+        "objects": (BROWN_MOREL, BROWN_OYSTER, GREEN_DEATHCAP_2, 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)
+            ),
+        ),
+        "nowrap": True,
+    },
+    "ForagaxTwoBiome-v4": {
+        "size": (15, 15),
+        "aperture_size": None,
+        "objects": (BROWN_MOREL_2, BROWN_OYSTER, GREEN_DEATHCAP_3, GREEN_FAKE_2),
+        "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)
+            ),
+        ),
+        "nowrap": True,
+    },
     "ForagaxTwoBiomeSmall-v1": {
         "size": (16, 8),
         "aperture_size": None,