continual-foragax 0.5.0__tar.gz → 0.6.0__tar.gz

continual_foragax-0.6.0/PKG-INFO

@@ -0,0 +1,16 @@
+Metadata-Version: 2.4
+Name: continual-foragax
+Version: 0.6.0
+Summary: A continual reinforcement learning benchmark
+Author-email: Steven Tang <stang5@ualberta.ca>
+Requires-Python: >=3.9
+Description-Content-Type: text/markdown
+Requires-Dist: gymnax
+Requires-Dist: six; python_version == "3.9"
+Provides-Extra: dev
+Requires-Dist: pre-commit>=4.3.0; extra == "dev"
+Requires-Dist: pytest>=8.4.1; extra == "dev"
+Requires-Dist: pytest-benchmark>=5.1.0; extra == "dev"
+Requires-Dist: ruff>=0.12.9; extra == "dev"
+
+# foragax

{continual_foragax-0.5.0 → continual_foragax-0.6.0}/pyproject.toml

@@ -1,17 +1,18 @@
 [project]
 name = "continual-foragax"
-version = "0.5.0"
+version = "0.6.0"
 description = "A continual reinforcement learning benchmark"
 readme = "README.md"
 authors = [
     { name = "Steven Tang", email = "stang5@ualberta.ca" }
 ]
-requires-python = ">=3.11"
+requires-python = ">=3.9"
 dependencies = [
-    "gymnax>=0.0.9",
+    "gymnax",
+    "six; python_version == '3.9'",
 ]
 
-[dependency-groups]
+[project.optional-dependencies]
 dev = [
     "pre-commit>=4.3.0",
     "pytest>=8.4.1",
@@ -29,6 +30,6 @@ build-backend = "setuptools.build_meta"
 [tool]
 [tool.commitizen]
 name = "cz_conventional_commits"
-version = "0.5.0"
+version = "0.6.0"
 tag_format = "$version"
 version_files = ["pyproject.toml"]

continual_foragax-0.6.0/src/continual_foragax.egg-info/PKG-INFO

@@ -0,0 +1,16 @@
+Metadata-Version: 2.4
+Name: continual-foragax
+Version: 0.6.0
+Summary: A continual reinforcement learning benchmark
+Author-email: Steven Tang <stang5@ualberta.ca>
+Requires-Python: >=3.9
+Description-Content-Type: text/markdown
+Requires-Dist: gymnax
+Requires-Dist: six; python_version == "3.9"
+Provides-Extra: dev
+Requires-Dist: pre-commit>=4.3.0; extra == "dev"
+Requires-Dist: pytest>=8.4.1; extra == "dev"
+Requires-Dist: pytest-benchmark>=5.1.0; extra == "dev"
+Requires-Dist: ruff>=0.12.9; extra == "dev"
+
+# foragax

continual_foragax-0.6.0/src/continual_foragax.egg-info/requires.txt

@@ -0,0 +1,10 @@
+gymnax
+
+[:python_version == "3.9"]
+six
+
+[dev]
+pre-commit>=4.3.0
+pytest>=8.4.1
+pytest-benchmark>=5.1.0
+ruff>=0.12.9

{continual_foragax-0.5.0 → continual_foragax-0.6.0}/src/foragax/env.py

@@ -6,7 +6,7 @@ Source: https://github.com/andnp/Foragax
 from dataclasses import dataclass
 from enum import IntEnum
 from functools import partial
-from typing import Any, Tuple
+from typing import Any, Tuple, Union
 
 import jax
 import jax.numpy as jnp
@@ -37,13 +37,13 @@ DIRECTIONS = jnp.array(
 class Biome:
     # Object generation frequencies for this biome
     object_frequencies: Tuple[float, ...] = ()
-    start: Tuple[int, int] | None = None
-    stop: Tuple[int, int] | None = None
+    start: Union[Tuple[int, int], None] = None
+    stop: Union[Tuple[int, int], None] = None
 
 
 @struct.dataclass
 class EnvParams(environment.EnvParams):
-    max_steps_in_episode: int | None
+    max_steps_in_episode: Union[int, None]
 
 
 @struct.dataclass
@@ -53,13 +53,13 @@ class EnvState(environment.EnvState):
     time: int
 
 
-class ForagaxEnv(environment.Environment[EnvState, EnvParams]):
+class ForagaxEnv(environment.Environment):
     """JAX implementation of Foragax environment."""
 
     def __init__(
         self,
-        size: Tuple[int, int] | int = (10, 10),
-        aperture_size: Tuple[int, int] | int = (5, 5),
+        size: Union[Tuple[int, int], int] = (10, 10),
+        aperture_size: Union[Tuple[int, int], int] = (5, 5),
         objects: Tuple[BaseForagaxObject, ...] = (),
         biomes: Tuple[Biome, ...] = (Biome(object_frequencies=()),),
     ):
@@ -103,7 +103,7 @@ class ForagaxEnv(environment.Environment[EnvState, EnvParams]):
         self,
         key: jax.Array,
         state: EnvState,
-        action: int | float | jax.Array,
+        action: Union[int, float, jax.Array],
         params: EnvParams,
     ) -> tuple[jax.Array, EnvState, jax.Array, jax.Array, dict[Any, Any]]:
         """Perform single timestep state transition."""

{continual_foragax-0.5.0 → continual_foragax-0.6.0}/src/foragax/registry.py

@@ -12,7 +12,7 @@ from foragax.env import (
 from foragax.objects import LARGE_MOREL, LARGE_OYSTER, MEDIUM_MOREL
 
 ENV_CONFIGS: Dict[str, Dict[str, Any]] = {
-    "ForagaxTwoBiomeSmall": {
+    "ForagaxTwoBiomeSmall-v1": {
         "size": (16, 8),
         "aperture_size": (5, 5),
         "objects": (LARGE_MOREL, LARGE_OYSTER),

{continual_foragax-0.5.0 → continual_foragax-0.6.0}/tests/test_foragax.py

@@ -246,8 +246,10 @@ def test_respawn():
     assert reward == FLOWER.reward_val
     assert state.object_grid[4, 3] < 0
 
+    steps_until_respawn = -state.object_grid[4, 3] // 2
+
     # Step until it respawns
-    for i in range(20):
+    for i in range(steps_until_respawn):
         key, step_key = jax.random.split(key)
         _, state, _, _, _ = env.step_env(step_key, state, Actions.UP, params)
         assert state.object_grid[4, 3] < 0

continual_foragax-0.5.0/PKG-INFO

@@ -1,10 +0,0 @@
-Metadata-Version: 2.4
-Name: continual-foragax
-Version: 0.5.0
-Summary: A continual reinforcement learning benchmark
-Author-email: Steven Tang <stang5@ualberta.ca>
-Requires-Python: >=3.11
-Description-Content-Type: text/markdown
-Requires-Dist: gymnax>=0.0.9
-
-# foragax

continual_foragax-0.5.0/src/continual_foragax.egg-info/PKG-INFO

@@ -1,10 +0,0 @@
-Metadata-Version: 2.4
-Name: continual-foragax
-Version: 0.5.0
-Summary: A continual reinforcement learning benchmark
-Author-email: Steven Tang <stang5@ualberta.ca>
-Requires-Python: >=3.11
-Description-Content-Type: text/markdown
-Requires-Dist: gymnax>=0.0.9
-
-# foragax

continual_foragax-0.5.0/src/continual_foragax.egg-info/requires.txt

@@ -1 +0,0 @@
-gymnax>=0.0.9