Commit fccd28b1 authored by Tiago de Freitas Pereira's avatar Tiago de Freitas Pereira
Browse files

Solving Issue #38 - BioDatabase method training_files causes tests to fail

parent e7722c0c
Pipeline #3809 passed with stages
in 62 minutes and 24 seconds
......@@ -12,7 +12,7 @@ import os
from bob.bio.base.database import BioDatabase, ZTBioDatabase
def check_database(database, groups=('dev',), protocol=None, training_depends=False, models_depend=False):
def check_database(database, groups=('dev',), protocol=None, training_depends=False, models_depend=False, skip_train=False):
assert isinstance(database, BioDatabase)
# load the directories
......@@ -23,8 +23,9 @@ def check_database(database, groups=('dev',), protocol=None, training_depends=Fa
if protocol is None: protocol = database.protocol
assert len(database.all_files()) > 0
assert len(database.training_files('train_extractor')) > 0
assert len(database.arrange_by_client(database.training_files('train_enroller'))) > 0
if not skip_train:
assert len(database.training_files('train_extractor')) > 0
assert len(database.arrange_by_client(database.training_files('train_enroller'))) > 0
for group in groups:
model_ids = database.model_ids_with_protocol(group, protocol=protocol)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment