lint

uber · Dec 16, 2023 · 34cddfc · 34cddfc
1 parent 71e4909
commit 34cddfc
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/tests/conftest.py b/tests/conftest.py
@@ -67,10 +67,9 @@ def _generate_data():
 
     yield _generate_data
 
+
 def pytest_addoption(parser):
-    parser.addoption(
-        "--runtf", action="store_true", default=False, help="run tf tests"
-    )
+    parser.addoption("--runtf", action="store_true", default=False, help="run tf tests")
 
 
 def pytest_configure(config):
@@ -85,4 +84,3 @@ def pytest_collection_modifyitems(config, items):
     for item in items:
         if "tf" in item.keywords:
             item.add_marker(skip_tf)
-
diff --git a/tests/test_dragonnet.py b/tests/test_dragonnet.py
@@ -3,6 +3,7 @@
 import shutil
 import pytest
 
+
 @pytest.mark.tf
 def test_save_load_dragonnet():
     y, X, w, tau, b, e = simulate_nuisance_and_easy_treatment(n=1000)