Minor fix to handle is_seg_dataset argument (#892)

Allows users to pass the is_seg_dataset, if desired. Otherwise, sets it to internal framework choice.

Author: anwai98

micro_sam/training/training.py

@@ -421,7 +421,7 @@ def default_sam_dataset(
     """
 
     # By default, let the 'default_segmentation_dataset' heuristic decide for itself.
-    is_seg_dataset = None
+    is_seg_dataset = kwargs.pop("is_seg_dataset", None)
 
     # Check if the raw inputs are RGB or not. If yes, use 'ImageCollectionDataset'.
     # Get valid raw paths to make checks possible.