From 70208939de60246b4b83821184276f22d942347c Mon Sep 17 00:00:00 2001 From: Keith Battocchi Date: Fri, 15 May 2020 10:53:18 -0400 Subject: [PATCH] Fix linting issues --- econml/tests/test_deepiv.py | 4 ++-- econml/tests/test_two_stage_least_squares.py | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/econml/tests/test_deepiv.py b/econml/tests/test_deepiv.py index bf235f20a..809c1efd3 100644 --- a/econml/tests/test_deepiv.py +++ b/econml/tests/test_deepiv.py @@ -481,8 +481,8 @@ def test_mog_models(self): l2 = keras.layers.Dense(10, activation='relu') l3 = keras.layers.Dense(10, activation='relu') - def norm(l): - return l # keras.layers.BatchNormalization() + def norm(lr): + return lr # keras.layers.BatchNormalization() x_network = l3(norm(l2(norm(l1(x_input))))) diff --git a/econml/tests/test_two_stage_least_squares.py b/econml/tests/test_two_stage_least_squares.py index f045af954..69c393735 100644 --- a/econml/tests/test_two_stage_least_squares.py +++ b/econml/tests/test_two_stage_least_squares.py @@ -127,8 +127,8 @@ def test_2sls(self): a = np.random.normal(size=(d_w, d_t)) b = np.random.normal(size=(d_w, d_y)) x = np.random.uniform(size=(n, d_x)) + e - p = x + z * e + w@a + np.random.uniform(size=(n, d_t)) - y = p * x + e + w@b + p = x + z * e + w @ a + np.random.uniform(size=(n, d_t)) + y = p * x + e + w @ b losses = [] marg_effs = []