Skip to content

Commit fe7107d

Browse files
committed
remove jax dependency and test on jax
Signed-off-by: Nathaniel <NathanielF@users.noreply.github.com>
1 parent 7f87402 commit fe7107d

File tree

3 files changed

+1
-4
lines changed

3 files changed

+1
-4
lines changed

causalpy/tests/test_integration_pymc_examples.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -504,7 +504,6 @@ def test_iv_reg():
504504
sample_kwargs=sample_kwargs
505505
),
506506
)
507-
result.model.sample_predictive_distribution(ppc_sampler="jax")
508507
result.model.sample_predictive_distribution(ppc_sampler="pymc")
509508
assert isinstance(df, pd.DataFrame)
510509
assert isinstance(data, pd.DataFrame)

environment.yml

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -15,4 +15,3 @@ dependencies:
1515
- seaborn>=0.11.2
1616
- statsmodels
1717
- xarray>=v2022.11.0
18-
- jax

pyproject.toml

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -39,8 +39,7 @@ dependencies = [
3939
"scipy",
4040
"seaborn>=0.11.2",
4141
"statsmodels",
42-
"xarray>=v2022.11.0",
43-
"jax"
42+
"xarray>=v2022.11.0"
4443
]
4544

4645
# List additional groups of dependencies here (e.g. development dependencies). Users

0 commit comments

Comments
 (0)