feat(workshop): clean up for workshop

guillaume-chervet · Dec 15, 2023 · 4f3c386 · 4f3c386
1 parent 3821f0b
commit 4f3c386
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/train/extraction_dataset.py b/train/extraction_dataset.py
@@ -47,7 +47,7 @@ def register_extracted_dataset(ml_client,
         extracted_images_dataset = Data(
             name=extracted_images_dataset_name,
             path=custom_extraction_path,
-            type=AssetTypes.CUSTOM_MODEL,
+            type=AssetTypes.URI_FOLDER,
             description="Extracted images for cats and dogs and others",
             version=str(version_dataset_extraction),
             tags={"hash": computed_hash, **tags},