pyRDDLGym-jax 2.7__tar.gz → 2.8__tar.gz

{pyrddlgym_jax-2.7 → pyrddlgym_jax-2.8}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pyRDDLGym-jax
-Version: 2.7
+Version: 2.8
 Summary: pyRDDLGym-jax: automatic differentiation for solving sequential planning problems in JAX.
 Home-page: https://github.com/pyrddlgym-project/pyRDDLGym-jax
 Author: Michael Gimelfarb, Ayal Taitler, Scott Sanner

pyrddlgym_jax-2.8/pyRDDLGym_jax/__init__.py

@@ -0,0 +1 @@
+__version__ = '2.8'

{pyrddlgym_jax-2.7 → pyrddlgym_jax-2.8}/pyRDDLGym_jax/examples/run_plan.py

@@ -25,36 +25,6 @@ from pyRDDLGym_jax.core.planner import (
     load_config, JaxBackpropPlanner, JaxOfflineController, JaxOnlineController
 )
 
-
-def run_cnn1d():
-    import haiku as hk
-    import jax
-    import jax.numpy as jnp
-
-    class CNN(hk.Module):
-        def __init__(self, name=None):
-            super().__init__(name=name)
-            self.conv1d_layer = hk.Conv1D(
-                output_channels=4,
-                kernel_shape=6,  # Kernel size for 1D convolution
-                padding="SAME",
-                name="conv"
-            )
-
-        def __call__(self, x):
-            return self.conv1d_layer(x)
-
-    # Example usage:
-    key = jax.random.PRNGKey(42)
-    input_data = jnp.ones([1, 4])  # Batch size 1, sequence length 10, 1 input channel
-
-    # Transform the Haiku module into a pure function
-    f = hk.transform(lambda x: CNN()(x))
-    params = f.init(key, input_data)
-    print(params['cnn/~/conv']['w'].shape)
-    print(params['cnn/~/conv']['b'].shape)
-    print(f.apply(params, key, input_data).shape)
-
     
 def main(domain: str, instance: str, method: str, episodes: int=1) -> None:
     
@@ -93,7 +63,6 @@ def main(domain: str, instance: str, method: str, episodes: int=1) -> None:
         
 
 def run_from_args(args):
-    run_cnn1d()
     if len(args) < 3:
         print('python run_plan.py <domain> <instance> <method> [<episodes>]')
         exit(1)

{pyrddlgym_jax-2.7 → pyrddlgym_jax-2.8}/pyRDDLGym_jax.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pyRDDLGym-jax
-Version: 2.7
+Version: 2.8
 Summary: pyRDDLGym-jax: automatic differentiation for solving sequential planning problems in JAX.
 Home-page: https://github.com/pyrddlgym-project/pyRDDLGym-jax
 Author: Michael Gimelfarb, Ayal Taitler, Scott Sanner

{pyrddlgym_jax-2.7 → pyrddlgym_jax-2.8}/setup.py

@@ -19,7 +19,7 @@ long_description = (Path(__file__).parent / "README.md").read_text()
 
 setup(
       name='pyRDDLGym-jax',
-      version='2.7',
+      version='2.8',
       author="Michael Gimelfarb, Ayal Taitler, Scott Sanner",
       author_email="mike.gimelfarb@mail.utoronto.ca, ataitler@gmail.com, ssanner@mie.utoronto.ca",
       description="pyRDDLGym-jax: automatic differentiation for solving sequential planning problems in JAX.",

pyrddlgym_jax-2.7/pyRDDLGym_jax/__init__.py

@@ -1 +0,0 @@
-__version__ = '2.7'