From 235f0e27da80b0068aabd3fc55ab85bf233bb2e2 Mon Sep 17 00:00:00 2001
From: mbsantiago <santiago.mbal@gmail.com>
Date: Tue, 24 Jun 2025 12:40:29 +0100
Subject: [PATCH] Add load dataset config function

---
 src/batdetect2/data/__init__.py | 2 ++
 src/batdetect2/data/datasets.py | 5 +++++
 2 files changed, 7 insertions(+)

diff --git a/src/batdetect2/data/__init__.py b/src/batdetect2/data/__init__.py
index 8d10f08..f9a5ee3 100644
--- a/src/batdetect2/data/__init__.py
+++ b/src/batdetect2/data/__init__.py
@@ -8,6 +8,7 @@ from batdetect2.data.annotations import (
 from batdetect2.data.datasets import (
     DatasetConfig,
     load_dataset,
+    load_dataset_config,
     load_dataset_from_config,
 )
 
@@ -19,5 +20,6 @@ __all__ = [
     "DatasetConfig",
     "load_annotated_dataset",
     "load_dataset",
+    "load_dataset_config",
     "load_dataset_from_config",
 ]
diff --git a/src/batdetect2/data/datasets.py b/src/batdetect2/data/datasets.py
index b960305..ab43b17 100644
--- a/src/batdetect2/data/datasets.py
+++ b/src/batdetect2/data/datasets.py
@@ -161,6 +161,11 @@ def insert_source_tag(
     )
 
 
+# TODO: add documentation
+def load_dataset_config(path: data.PathLike, field: Optional[str] = None):
+    return load_config(path=path, schema=DatasetConfig, field=field)
+
+
 def load_dataset_from_config(
     path: data.PathLike,
     field: Optional[str] = None,