Fix linting issues

py-why · May 15, 2020 · 7020893 · 7020893
1 parent 54f531b
commit 7020893
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git 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
@@ -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 = []