Fix linting issues

Author: kbattocchi

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)))))
 

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 = []