Remove old pyro and whynot code

bradyneal · Mar 29, 2021 · 2426b1a · 2426b1a
1 parent 1dcaef1
commit 2426b1a
Show file tree

Hide file tree

Showing 6 changed files with 1 addition and 208 deletions.
diff --git a/data/__init__.py b/data/__init__.py
@@ -1,2 +1 @@
 from .synthetic import *
-from .whynot_simulators import *
diff --git a/data/whynot_simulators.py b/data/whynot_simulators.py
diff --git a/models/__init__.py b/models/__init__.py
@@ -1,6 +1,4 @@
 from .linear import *
-from .linear_pyro import *
-from .vae import *
 from .nonlinear import *
 from .tarnet import *
-from .gp import *
+from .gp import *
diff --git a/models/linear_pyro.py b/models/linear_pyro.py
diff --git a/models/vae.py b/models/vae.py
diff --git a/tests/test_data.py b/tests/test_data.py
@@ -5,7 +5,6 @@
 import torch
 
 from data.synthetic import generate_wty_linear_scalar_data, generate_wty_linear_multi_w_data
-from data.whynot_simulators import generate_lalonde_random_outcome
 from data.lalonde import load_lalonde
 from data.ihdp import load_ihdp_datasets
 from utils import NUMPY, PANDAS, PANDAS_SINGLE, TORCH
@@ -184,16 +183,6 @@ def test_lalonde_original_cps3():
     assert round(ate) == -1008
 
 
-def test_lalonde_random_outcome_data_pandas():
-    (w, t, y), causal_effects = generate_lalonde_random_outcome(data_format=PANDAS)
-    assert isinstance(w, pd.DataFrame) and isinstance(t, pd.Series) and isinstance(y, pd.Series)
-
-
-def test_lalonde_random_outcome_data_torch():
-    (w, t, y), causal_effects = generate_lalonde_random_outcome(data_format=TORCH)
-    assert all(isinstance(x, torch.Tensor) for x in (w, t, y))
-
-
 def test_multivariate_w_data():
     n = 10
     d = 5
Original file line number	Diff line number	Diff line change
		@@ -1,2 +1 @@
		from .synthetic import *
		from .whynot_simulators import *