Skip to content

Commit 428d7db

Browse files
committed
friendlyify formats for new export route
1 parent fa893ba commit 428d7db

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

roboflow/core/version.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -154,6 +154,8 @@ def download(self, model_format=None, location=None):
154154

155155
self.__wait_if_generating()
156156

157+
model_format = self.__get_format_identifier(model_format)
158+
157159
if model_format not in self.exports:
158160
self.export(model_format)
159161

@@ -162,8 +164,6 @@ def download(self, model_format=None, location=None):
162164
if location is None:
163165
location = self.__get_download_location()
164166

165-
model_format = self.__get_format_identifier(model_format)
166-
167167
if self.__api_key == "coco-128-sample":
168168
link = "https://app.roboflow.com/ds/n9QwXwUK42?key=NnVCe2yMxP"
169169
else:
@@ -194,6 +194,8 @@ def export(self, model_format=None):
194194
:raises RuntimeError / HTTPError:
195195
"""
196196

197+
model_format = self.__get_format_identifier(model_format)
198+
197199
self.__wait_if_generating()
198200

199201
url = self.__get_download_url(model_format)

0 commit comments

Comments
 (0)