lionel/batdetect2
1
0
Fork 0
mirror of https://github.com/macaodha/batdetect2.git synced 2025-06-29 14:41:58 +02:00
Code Issues Actions Packages Projects Releases Wiki Activity

Add load dataset config function

Browse Source
This commit is contained in:
mbsantiago 2025-06-24 12:40:29 +01:00
parent b5b4229990
commit 235f0e27da
2 changed files with 7 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/batdetect2/data/__init__.py
View File

@ -8,6 +8,7 @@ from batdetect2.data.annotations import (
from batdetect2.data.datasets import ( from batdetect2.data.datasets import (
DatasetConfig, DatasetConfig,
load_dataset, load_dataset,
load_dataset_config,
load_dataset_from_config, load_dataset_from_config,
) )
@ -19,5 +20,6 @@ __all__ = [
"DatasetConfig", "DatasetConfig",
"load_annotated_dataset", "load_annotated_dataset",
"load_dataset", "load_dataset",
"load_dataset_config",
"load_dataset_from_config", "load_dataset_from_config",
] ]

5
src/batdetect2/data/datasets.py
View File

@ -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( def load_dataset_from_config(
path: data.PathLike, path: data.PathLike,
field: Optional[str] = None, field: Optional[str] = None,