diff --git a/src/bob/pipelines/dataset/database.py b/src/bob/pipelines/dataset/database.py index 93da39b6934468e8a6191f17d1eef595530cb987..2ea4653cdf410d24ec7d73d804e0a7788974090d 100644 --- a/src/bob/pipelines/dataset/database.py +++ b/src/bob/pipelines/dataset/database.py @@ -245,8 +245,8 @@ class FileListDatabase: # Save to bob_data/protocols, or if present, in a category sub directory. subdir = Path("protocols") - if hasattr(cls, "dataset_protocols_category"): - subdir = subdir / getattr(cls, "dataset_protocols_category") + if hasattr(cls, "category"): + subdir = subdir / getattr(cls, "category") # Retrieve the file from the server (or use the local version). return retrieve_protocols( diff --git a/tests/test_database.py b/tests/test_database.py index f4f68d72f5073a7282b56f567ca89da157e2252c..b840546bac0c3a7ec7a93e8030213dd13a314dc8 100644 --- a/tests/test_database.py +++ b/tests/test_database.py @@ -122,11 +122,11 @@ def test_filelist_class(monkeypatch): class DBDownloadCustomCategory(FileListDatabase): name = "atnt" + category = desired_category dataset_protocols_checksum = "f529acef" dataset_protocols_urls = [ "https://www.idiap.ch/software/bob/databases/latest/base/atnt-f529acef.tar.gz" ] - dataset_protocols_category = desired_category assert DBDownloadCustomCategory.protocols() == ["idiap_protocol"] assert (