Skip to content

Commit

Permalink
Follows code style
Browse files Browse the repository at this point in the history
  • Loading branch information
RohitRathore1 authored and tom-andersson committed Feb 2, 2024
1 parent 6a80fc4 commit aeccc09
Showing 1 changed file with 6 additions and 2 deletions.
8 changes: 6 additions & 2 deletions tests/test_active_learning.py
Original file line number Diff line number Diff line change
Expand Up @@ -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,
Expand Down

0 comments on commit aeccc09

Please sign in to comment.