Update code

cvat-ai · Mar 13, 2024 · 8871081 · 8871081
1 parent 4c767ad
commit 8871081
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/datumaro/plugins/synthetic_data/image_generator.py b/datumaro/plugins/synthetic_data/image_generator.py
@@ -67,8 +67,9 @@ def generate_dataset(self) -> None:
         mp_ctx = get_context("spawn")  # On Mac 10.15 and Python 3.7 fork leads to hangs
         with mp_ctx.Pool(processes=self._cpu_count) as pool:
             params = pool.map(
-                self._generate_category, [Random(i) for i in range(self._categories)]
-            )  # nosec - disable B311
+                self._generate_category,
+                [Random(i) for i in range(self._categories)],  # nosec - disable B311
+            )
 
         instances_weights = np.repeat(self._weights, self._instances, axis=0)
         weight_per_img = np.tile(instances_weights, (self._categories, 1))