Skip to content

Commit 5336a86

Browse files
authored
Merge pull request #121 from roboflow/fix/118_bug_when_ultralytics_is_installed_in_env_and_dataset_is_being_downloaded_as_coco_2
fix/118_bug_when_ultralytics_is_installed_in_env_and_dataset_is_being_downloaded_as_coco_2
2 parents 34cb8c8 + 9148441 commit 5336a86

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

roboflow/core/version.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -570,7 +570,7 @@ def __reformat_yaml(self, location: str, format: str):
570570
"""
571571
data_path = os.path.join(location, "data.yaml")
572572

573-
def callback(content: dict) -> dict:
573+
def data_yaml_callback(content: dict) -> dict:
574574
if format == "mt-yolov6":
575575
content["train"] = location + content["train"].lstrip(".")
576576
content["val"] = location + content["val"].lstrip(".")
@@ -590,7 +590,8 @@ def callback(content: dict) -> dict:
590590
pass
591591
return content
592592

593-
amend_data_yaml(path=data_path, callback=callback)
593+
if format in ["yolov5pytorch", "mt-yolov6", "yolov7pytorch", "yolov8"]:
594+
amend_data_yaml(path=data_path, callback=data_yaml_callback)
594595

595596
def __str__(self):
596597
"""string representation of version object."""

0 commit comments

Comments
 (0)