safenax 0.4.7__py3-none-any.whl → 0.4.9__py3-none-any.whl

safenax/portfolio_optimization/po_garch.py

@@ -242,8 +242,8 @@ class PortfolioOptimizationGARCH(Environment):
     def action_space(self, params: EnvParams) -> spaces.Box:
         """Action space: portfolio weights (including cash)."""
         return spaces.Box(
-            low=-jnp.inf,
-            high=jnp.inf,
+            low=-10.0,
+            high=10.0,
             shape=(self.num_assets + 1,),  # +1 for cash
             dtype=jnp.float32,
         )

safenax/wrappers/log.py

@@ -28,7 +28,7 @@ class LogWrapper(GymnaxWrapper):
         self, key: jax.Array, params: environment.EnvParams | None = None
     ) -> tuple[jax.Array, LogEnvState]:
         obs, env_state = self._env.reset(key, params)
-        state = LogEnvState(env_state, 0, 0, 0, 0, 0, 0)
+        state = LogEnvState(env_state, 0.0, 0.0, 0, 0.0, 0.0, 0)
         return obs, state
 
     @partial(jax.jit, static_argnames=("self",))
@@ -60,13 +60,13 @@ class LogWrapper(GymnaxWrapper):
         new_episode_length = state.episode_lengths + 1
         state = LogEnvState(
             env_state=env_state,
-            episode_returns=new_episode_return * (1 - done),
-            episode_cost_returns=new_episode_cost_return * (1 - done),
+            episode_returns=new_episode_return * (1.0 - done),
+            episode_cost_returns=new_episode_cost_return * (1.0 - done),
             episode_lengths=new_episode_length * (1 - done),
-            returned_episode_returns=state.returned_episode_returns * (1 - done)
+            returned_episode_returns=state.returned_episode_returns * (1.0 - done)
             + new_episode_return * done,
             returned_episode_cost_returns=state.returned_episode_cost_returns
-            * (1 - done)
+            * (1.0 - done)
             + new_episode_cost_return * done,
             returned_episode_lengths=state.returned_episode_lengths * (1 - done)
             + new_episode_length * done,

{safenax-0.4.7.dist-info → safenax-0.4.9.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: safenax
-Version: 0.4.7
+Version: 0.4.9
 Summary: Constrained environments with a gymnax interface
 Project-URL: Homepage, https://github.com/0xprofessooor/safenax
 Project-URL: Repository, https://github.com/0xprofessooor/safenax

{safenax-0.4.7.dist-info → safenax-0.4.9.dist-info}/RECORD

@@ -8,11 +8,11 @@ safenax/frozen_lake/frozen_lake_v1.py,sha256=6Yy9tm4MbrNiYXDNi091Jsh9iwoEcKz3TrO
 safenax/frozen_lake/frozen_lake_v2.py,sha256=rXSJ8lJKPsF0ZVQcQpBm6ysBYq5I2G0GIfsfLD1-TmA,9731
 safenax/portfolio_optimization/__init__.py,sha256=tbtCF4fVfan2nfFJc2wNl24hCALSb0yON1OYboN5OGk,245
 safenax/portfolio_optimization/po_crypto.py,sha256=Bi4QCd4MoeQAnhag22MFWdqy1uQ5hVQdiwYymP9v7N4,7342
-safenax/portfolio_optimization/po_garch.py,sha256=qIVWDJg1qw95pSqVViwQfZGqZ66NNAGyIa337qyMmbU,34690
+safenax/portfolio_optimization/po_garch.py,sha256=ROI_rrUGqYhRsqtT2t60X7WkreJe6a2imf9ZIyu8mqo,34684
 safenax/wrappers/__init__.py,sha256=v9wyHyR482ZEfmfTtcGabpf_lUHze4fy-NjrEaGv3zA,158
 safenax/wrappers/brax.py,sha256=svijcYVoWy5ej7RRLuN8VixDL_cMXKBK-veFsC57LRE,2985
-safenax/wrappers/log.py,sha256=jsjT0FJBo21rCM6D2Hx9fOwXLdwP1MW6PAx1BJBP2lA,2842
-safenax-0.4.7.dist-info/METADATA,sha256=qYYCkUAu6eBJtMJw8uhM6aEQvSrJ3YMC2SjnsWHoCh0,1202
-safenax-0.4.7.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-safenax-0.4.7.dist-info/licenses/LICENSE,sha256=BI7P9lDrJUcIUIX_4sCSE9pKHgCYIKWzHCOFyn85eKk,1077
-safenax-0.4.7.dist-info/RECORD,,
+safenax/wrappers/log.py,sha256=npAK7SBguy_aDucqxJ47p0ywINi6MmN3NjC5TklfS4k,2858
+safenax-0.4.9.dist-info/METADATA,sha256=ACptVPfLg1PwXJFITsDLTobef0EeYUgaip0XBBN8eY0,1202
+safenax-0.4.9.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+safenax-0.4.9.dist-info/licenses/LICENSE,sha256=BI7P9lDrJUcIUIX_4sCSE9pKHgCYIKWzHCOFyn85eKk,1077
+safenax-0.4.9.dist-info/RECORD,,