fix(label_split_data): add suffle

guillaume-chervet · Jan 12, 2024 · 767dc8c · 767dc8c
1 parent 5a87f4a
commit 767dc8c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/train/extraction/command.py b/train/extraction/command.py
@@ -16,7 +16,7 @@
 hash_output = args.hash_output
 
 extract_images = ExtractImages(DataManager())
-result = extract_images(pdfs_input, images_output)
+result = extract_images.extract_images(pdfs_input, images_output)
 
 computed_hash = hash_dir(images_output)
 with open(str(Path(hash_output) / "hash.txt"), "w") as file: