Merge branch 'dev' of https://github.com/Project-AgML/AgML into dev

Project-AgML · Oct 14, 2024 · dc52570 · dc52570
2 parents fe6bf95 + 129d92a
commit dc52570
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/agml/__init__.py b/agml/__init__.py
@@ -12,7 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-__version__ = '0.6.1'
+__version__ = '0.6.2'
 __all__ = ['data', 'synthetic', 'backend', 'viz', 'io']
 
 

diff --git a/agml/data/exporters/yolo.py b/agml/data/exporters/yolo.py
@@ -154,7 +154,7 @@ def export_yolo(dataset, yolo_path=None):
     """.format(yolo_data_path, os.path.join(yolo_path, 'data', 'dataset.yaml'))
     log(information)
 
-    return yolo_data_path
+    return {'dataset_path': yolo_data_path, 'metadata_path': os.path.join(yolo_path, 'data', 'dataset.yaml')}