lint

Author: Samoed

tests/data/test_data_handler.py

@@ -233,10 +233,12 @@ def test_few_shot_split(dataset):
     desired_specs = {
         "train_0": {0: 2, 1: 2, 2: 2, 3: 2},
         "train_1": {2: 2, 0: 2, None: 2, 1: 1, 3: 1},
-        "validation_0": {0: 3, 1: 4, 2: 3, 3: 4, },
+        "validation_0": {0: 3, 1: 4, 2: 3, 3: 4},
         "validation_1": {None: 14, 3: 1, 0: 1, 1: 1, 2: 1},
         "test": {None: 4, 0: 2, 2: 2, 3: 2, 1: 2},
     }
 
     for data_split in dh.dataset:
-        assert Counter(dh.dataset[data_split][dh.dataset.label_feature]) == desired_specs[data_split], f"Failed for {data_split}"
+        assert (
+            Counter(dh.dataset[data_split][dh.dataset.label_feature]) == desired_specs[data_split]
+        ), f"Failed for {data_split}"