continual-foragax 0.41.0__tar.gz → 0.42.1__tar.gz

{continual_foragax-0.41.0 → continual_foragax-0.42.1}/PKG-INFO

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

{continual_foragax-0.41.0 → continual_foragax-0.42.1}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "continual-foragax"
-version = "0.41.0"
+version = "0.42.1"
 description = "A continual reinforcement learning benchmark"
 readme = "README.md"
 authors = [
@@ -12,6 +12,10 @@ dependencies = [
     "six; python_version < '3.10'",
 ]
 
+[tool.uv]
+index-strategy = "unsafe-best-match"
+find-links = ["https://storage.googleapis.com/jax-releases/jax_releases.html"]
+
 [dependency-groups]
 dev = [
     "pre-commit",
@@ -30,7 +34,7 @@ build-backend = "setuptools.build_meta"
 [tool]
 [tool.commitizen]
 name = "cz_conventional_commits"
-version = "0.41.0"
+version = "0.42.1"
 tag_format = "$version"
 version_files = ["pyproject.toml"]
 

{continual_foragax-0.41.0 → continual_foragax-0.42.1}/src/continual_foragax.egg-info/PKG-INFO

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

{continual_foragax-0.41.0 → continual_foragax-0.42.1}/src/foragax/registry.py

@@ -154,6 +154,17 @@ ENV_CONFIGS: Dict[str, Dict[str, Any]] = {
         "biome_consumption_threshold": 1000,
         "dynamic_biome_spawn_empty": 0.4,
     },
+    "ForagaxDiwali-v6": {
+        "size": (15, 15),
+        "aperture_size": None,
+        "objects": None,
+        "biomes": (Biome(start=(0, 0), stop=(15, 15), object_frequencies=(0.4,)),),
+        "nowrap": False,
+        "deterministic_spawn": True,
+        "dynamic_biomes": True,
+        "biome_consumption_threshold": 10000,
+        "dynamic_biome_spawn_empty": 0.4,
+    },
     "ForagaxTwoBiome-v1": {
         "size": (15, 15),
         "aperture_size": None,
@@ -615,6 +626,13 @@ def make(
             regen_delay=(9, 11),
             reward_repeat=100,
         )[:1]
+    if env_id == "ForagaxDiwali-v6":
+        config["objects"] = create_fourier_objects(
+            num_fourier_terms=10,
+            reward_delay=reward_delay,
+            regen_delay=(9, 11),
+            reward_repeat=1000,
+        )[:1]
 
     if env_id == "ForagaxSineTwoBiome-v1":
         biome1_oyster, biome1_deathcap, biome2_oyster, biome2_deathcap = (