continual-foragax 0.5.0__py3-none-any.whl → 0.7.0__py3-none-any.whl

continual_foragax-0.7.0.dist-info/METADATA

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

continual_foragax-0.7.0.dist-info/RECORD

@@ -0,0 +1,9 @@
+foragax/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+foragax/env.py,sha256=ykSjKvm5zeO_24Y1tsWWwAoUCkxRQjzj_QdXjsZkSQI,15188
+foragax/objects.py,sha256=j0sQ_qAjx_tGu4Xt7SXmwd9hsEHJXAGFY5I5VIf9_SM,4000
+foragax/registry.py,sha256=ISaqGMCy-IL9QCk7pmbYw59QVaSurKgLK8RS4wzf-pg,2187
+continual_foragax-0.7.0.dist-info/METADATA,sha256=DM4D9vZxvv3PgjA2kbq2_4NphitquabxuQoj-GvO_Ug,486
+continual_foragax-0.7.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+continual_foragax-0.7.0.dist-info/entry_points.txt,sha256=Qiu6iE_XudrDO_bVAMeA435h4PO9ourt8huvSHiuMPc,41
+continual_foragax-0.7.0.dist-info/top_level.txt,sha256=-z3SDK6RfLIcLI24n8rdbeFzlVY3hunChzlu-v1Fncs,8
+continual_foragax-0.7.0.dist-info/RECORD,,

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, Dict, 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,9 +103,9 @@ 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]]:
+    ) -> Tuple[jax.Array, EnvState, jax.Array, jax.Array, Dict[Any, Any]]:
         """Perform single timestep state transition."""
         num_obj_types = len(self.object_ids)
         # Decode the object grid: positive values are objects, negative are timers (treat as empty)
@@ -167,7 +167,7 @@ class ForagaxEnv(environment.Environment[EnvState, EnvParams]):
 
     def reset_env(
         self, key: jax.Array, params: EnvParams
-    ) -> tuple[jax.Array, EnvState]:
+    ) -> Tuple[jax.Array, EnvState]:
         """Reset environment state."""
         key, subkey = jax.random.split(key)
 

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.dist-info/METADATA

@@ -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.dist-info/RECORD

@@ -1,9 +0,0 @@
-foragax/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-foragax/env.py,sha256=m1LcFyi-RBCViEiaF3LqQ_K-2vy4yu8nTK1rujZ8f4A,15161
-foragax/objects.py,sha256=j0sQ_qAjx_tGu4Xt7SXmwd9hsEHJXAGFY5I5VIf9_SM,4000
-foragax/registry.py,sha256=IpoE5iiHXQIcIwdAr0fXjOwiy9qyWF7DBnDGnGAFUSI,2184
-continual_foragax-0.5.0.dist-info/METADATA,sha256=Me0ilQ4e3nLSruu4-_c4sNXR8_CS_PwgE03u-uzu7Kc,266
-continual_foragax-0.5.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-continual_foragax-0.5.0.dist-info/entry_points.txt,sha256=Qiu6iE_XudrDO_bVAMeA435h4PO9ourt8huvSHiuMPc,41
-continual_foragax-0.5.0.dist-info/top_level.txt,sha256=-z3SDK6RfLIcLI24n8rdbeFzlVY3hunChzlu-v1Fncs,8
-continual_foragax-0.5.0.dist-info/RECORD,,