fix linting

open-mmlab · Mar 18, 2024 · ffbee78 · ffbee78
1 parent bc954df
commit ffbee78
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/tools/misc/browse_dataset.py b/tools/misc/browse_dataset.py
@@ -112,14 +112,18 @@ def main():
         def generate_index_generator(dataset_starting_indexes: list,
                                      max_item_datasets: int):
             for relative_idx in range(max(max_item_datasets)):
-                for dataset_idx, dataset_starting_idx in enumerate(dataset_starting_indexes):
+                for dataset_idx, dataset_starting_idx in enumerate(
+                        dataset_starting_indexes):
                     if relative_idx >= max_item_datasets[dataset_idx]:
                         continue
                     yield dataset_starting_idx + relative_idx
 
         # Generate starting indexes for each dataset
         dataset_starting_indexes = list(accumulate([0] + dataset.lens[:-1]))
-        max_item_datasets = [min(dataset_len, args.max_item_per_dataset) for dataset_len in dataset.lens]
+        max_item_datasets = [
+            min(dataset_len, args.max_item_per_dataset)
+            for dataset_len in dataset.lens
+        ]
 
         # Generate indexes using the generator
         indexes = generate_index_generator(dataset_starting_indexes,