remove jax dependency and test on jax

NathanielF · NathanielF · commit fe7107dc3d13 · 2024-06-13T17:55:39.000+01:00
Signed-off-by: Nathaniel &lt;NathanielF@users.noreply.github.com&gt;
diff --git a/causalpy/tests/test_integration_pymc_examples.py b/causalpy/tests/test_integration_pymc_examples.py
@@ -504,7 +504,6 @@ def test_iv_reg():
             sample_kwargs=sample_kwargs
         ),
     )
-    result.model.sample_predictive_distribution(ppc_sampler="jax")
     result.model.sample_predictive_distribution(ppc_sampler="pymc")
     assert isinstance(df, pd.DataFrame)
     assert isinstance(data, pd.DataFrame)
diff --git a/environment.yml b/environment.yml
@@ -15,4 +15,3 @@ dependencies:
   - seaborn>=0.11.2
   - statsmodels
   - xarray>=v2022.11.0
-  - jax
diff --git a/pyproject.toml b/pyproject.toml
@@ -39,8 +39,7 @@ dependencies = [
     "scipy",
     "seaborn>=0.11.2",
     "statsmodels",
-    "xarray>=v2022.11.0",
-    "jax"
+    "xarray>=v2022.11.0"
 ]
 
 # List additional groups of dependencies here (e.g. development dependencies). Users

Original file line number	Diff line number	Diff line change
`@@ -504,7 +504,6 @@ def test_iv_reg():`
`504`	`504`	`sample_kwargs=sample_kwargs`
`505`	`505`	`),`
`506`	`506`	`)`
`507`		`- result.model.sample_predictive_distribution(ppc_sampler="jax")`
`508`	`507`	`result.model.sample_predictive_distribution(ppc_sampler="pymc")`
`509`	`508`	`assert isinstance(df, pd.DataFrame)`
`510`	`509`	`assert isinstance(data, pd.DataFrame)`
Original file line number	Diff line number	Diff line change
`@@ -39,8 +39,7 @@ dependencies = [`
`39`	`39`	`"scipy",`
`40`	`40`	`"seaborn>=0.11.2",`
`41`	`41`	`"statsmodels",`
`42`		`- "xarray>=v2022.11.0",`
`43`		`- "jax"`
	`42`	`+ "xarray>=v2022.11.0"`
`44`	`43`	`]`
`45`	`44`
`46`	`45`	`# List additional groups of dependencies here (e.g. development dependencies). Users`