format: black and ruff

luxonis · Apr 8, 2024 · 140ecf3 · 140ecf3
1 parent 00962b2
commit 140ecf3
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/datadreamer/pipelines/generate_dataset_from_scratch.py b/datadreamer/pipelines/generate_dataset_from_scratch.py
@@ -579,7 +579,11 @@ def main():
     # Convert annotations to LuxonisDataset format
     if args.dataset_format == "luxonis-dataset":
         convert_dataset.convert_dataset(
-            args.save_dir, args.save_dir, "luxonis-dataset", args.split_ratios, copy_files=False
+            args.save_dir,
+            args.save_dir,
+            "luxonis-dataset",
+            args.split_ratios,
+            copy_files=False,
         )
 
 

diff --git a/datadreamer/utils/dataset_utils.py b/datadreamer/utils/dataset_utils.py
@@ -1,9 +1,6 @@
 import json
 import os
 
-from luxonis_ml.data import LuxonisDataset
-from PIL import Image
-
 
 def save_annotations_to_json(
     image_paths,
@@ -31,4 +28,3 @@ def save_annotations_to_json(
     # Save to JSON file
     with open(os.path.join(save_dir, file_name), "w") as f:
         json.dump(annotations, f, indent=4)
-