black

tests/conftest.py

@@ -1,19 +1,23 @@
 import pytest
 from pathlib import Path
 
-def pytest_collection_modifyitems(config, items):    
+
+def pytest_collection_modifyitems(config, items):
     if not config.getoption("--integration"):
-        skip_integration = pytest.mark.skip(reason="Integration test - use --integration to run")
+        skip_integration = pytest.mark.skip(
+            reason="Integration test - use --integration to run"
+        )
         for item in items:
             test_path = Path(item.fspath)
             if "integration" in test_path.parts:
                 item.add_marker(skip_integration)
 
+
 def pytest_addoption(parser):
     print("Adding custom option")  # Debug print
     parser.addoption(
         "--integration",
         action="store_true",
         default=False,
-        help="run integration tests"
-    )
+        help="run integration tests",
+    )

tests/integration/test_transfer_load.py

@@ -3,6 +3,7 @@
 import pandas as pd
 from baskerville import seqnn
 
+
 def test_housby():
     model_file = "/home/yuanh/analysis/Borzoi_transfer/exp_10_10_23/hayflick/houlsby/latent_8/train/f0c0/train/model_best.h5"
     targets_file = "tests/data/transfer/targets.txt"