feat: ImageDataset name is now optional (#20)

bencevans · Jun 23, 2022 · cb841bc · cb841bc
1 parent 5d0c265
commit cb841bc
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/camtrapml/dataset.py b/camtrapml/dataset.py
@@ -6,7 +6,7 @@
 from os import walk
 
 from camtrapml.image.utils import is_image
-
+from typing import Union
 
 class ImageDataset:
     """
@@ -15,7 +15,7 @@ class ImageDataset:
     name: str
     path: Path
 
-    def __init__(self, name: str, path: Path):
+    def __init__(self,path: Path, name : Union[None, str] = None):
         self.name = name
         self.path = Path(path).expanduser()
         self.image_set = None