Follows code style

alan-turing-institute · Feb 2, 2024 · aeccc09 · aeccc09
1 parent 6a80fc4
commit aeccc09
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/tests/test_active_learning.py b/tests/test_active_learning.py
@@ -198,9 +198,13 @@ def test_greedy_algorithm_column_names(self):
         X_new_df, acquisition_fn_ds = alg(acquisition_fn, task)
 
         # Assert
-        expected_columns = ['lat', 'lon']  # Replace with actual expected column names
+        expected_columns = ["lat", "lon"]  # Replace with actual expected column names
         actual_columns = X_new_df.columns.tolist()
-        self.assertEqual(expected_columns, actual_columns, "Column names do not match the expected names")
+        self.assertEqual(
+            expected_columns,
+            actual_columns,
+            "Column names do not match the expected names",
+        )
 
     def test_greedy_alg_with_aux_at_targets_without_task_loader_raises_value_error(
         self,