clean up lint

JacksonBurns · Jan 25, 2024 · ec8346d · ec8346d
1 parent dfa9f4a
commit ec8346d
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/fastprop/fastprop_core.py b/fastprop/fastprop_core.py
@@ -211,11 +211,11 @@ def __init__(self, number_features, target_scaler, num_epochs, hidden_size, lear
     def get_metrics(problem_type):
         if problem_type == "regression":
             return "mse", "rmse"
-        elif  problem_type == "multilabel":
+        elif problem_type == "multilabel":
             return "bce", "auroc"
-        elif  problem_type == "multiclass":
+        elif problem_type == "multiclass":
             return "kldiv", "auroc"
-        elif  problem_type == "binary":
+        elif problem_type == "binary":
             return "bce", "accuracy"
         else:
             raise RuntimeError(f"Unsupported problem type '{problem_type}'!")