lint

stanfordmlgroup · Jan 9, 2024 · 0f4bbfa · 0f4bbfa
1 parent 046743c
commit 0f4bbfa
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/tests/test_distns.py b/tests/test_distns.py
@@ -51,9 +51,11 @@ def classification_data():
     )
     return X_train, X_test, y_train, y_test
 
+
 def is_t_distribution(dist, learner, regression_data):
     return dist == T
 
+
 @pytest.mark.slow
 @pytest.mark.parametrize(
     "dist",
@@ -66,7 +68,9 @@ def is_t_distribution(dist, learner, regression_data):
         DecisionTreeRegressor(criterion="friedman_mse", max_depth=5),
     ],
 )
-@pytest.mark.xfail(condition=is_t_distribution, reason="Known to fail with T distribution")
+@pytest.mark.xfail(
+    condition=is_t_distribution, reason="Known to fail with T distribution"
+)
 def test_dists_runs_on_examples_logscore(dist: Distn, learner, regression_data):
     X_train, X_test, y_train, y_test = regression_data
     # TODO: test early stopping features