Merge pull request #96 from roboflow/feature/add_dataset_download_overwrite_flag

add dataset download overwrite flag

Author: Jacobsolawetz

roboflow/__init__.py

@@ -8,7 +8,7 @@
 from roboflow.core.project import Project
 from roboflow.core.workspace import Workspace
 
-__version__ = "0.2.24"
+__version__ = "0.2.25"
 
 
 def check_key(api_key, model, notebook, num_retries=0):

roboflow/core/version.py

@@ -144,12 +144,13 @@ def __wait_if_generating(self, recurse=False):
                 sys.stdout.flush()
             return
 
-    def download(self, model_format=None, location=None):
+    def download(self, model_format=None, location=None, overwrite: bool = True):
         """
         Download and extract a ZIP of a version's dataset in a given format
 
         :param model_format: A format to use for downloading
         :param location: An optional path for saving the file
+        :param overwrite: An optional flag to prevent dataset overwrite when dataset is already downloaded
 
         :return: Dataset
         """
@@ -165,6 +166,10 @@ def download(self, model_format=None, location=None):
 
         if location is None:
             location = self.__get_download_location()
+        if os.path.exists(location) and not overwrite:
+            return Dataset(
+                self.name, self.version, model_format, os.path.abspath(location)
+            )
 
         if self.__api_key == "coco-128-sample":
             link = "https://app.roboflow.com/ds/n9QwXwUK42?key=NnVCe2yMxP"