Skip to content

Commit fdc30df

Browse files
committed
return zip_file_name for yolonas
1 parent 1b1f988 commit fdc30df

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

roboflow/core/version.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -781,7 +781,7 @@ def _process_yolonas(self, model_type: str, model_path: str, filename: str = "we
781781
if file in ["model_artifacts.json", filename]:
782782
raise (ValueError(f"File {file} not found. Please make sure to provide a" " valid model path."))
783783

784-
self.upload_zip(model_type, model_path, zip_file_name)
784+
return zip_file_name
785785

786786
def _upload_zip(self, model_type: str, model_path: str, model_file_name: str):
787787
res = requests.get(

0 commit comments

Comments
 (0)