Add pycodestyle warnings

Signed-off-by: Keith Battocchi <[email protected]>
ronikobrosly · Aug 11, 2024 · f52f422 · f52f422
1 parent f9ce0d2
commit f52f422
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/notebooks/Metalearners Examples.ipynb b/notebooks/Metalearners Examples.ipynb
@@ -109,7 +109,7 @@
     "# Define DGP\n",
     "def generate_data(n, d, controls_outcome, treatment_effect, propensity):\n",
     "    \"\"\"Generate population data for given untreated_outcome, treatment_effect and propensity functions.\n",
-    "    \n",
+    "\n",
     "    Parameters\n",
     "    ----------\n",
     "        n (int): population size\n",

diff --git a/pyproject.toml b/pyproject.toml
@@ -155,6 +155,7 @@ ignore = [
 extend-select = [
     "D", # Docstring
     "E501", # Line too long
+    "W", # Pycodestyle warnings
 ]
 extend-per-file-ignores = { "econml/tests" = ["D"] } # ignore docstring rules for tests