Commit f96f0fb4 authored by Olegs NIKISINS's avatar Olegs NIKISINS

Fixed the HLDI of the putvein db - conversion of group names

parent 3949767a
Pipeline #13215 failed with stages
in 3 minutes 53 seconds
......@@ -80,6 +80,14 @@ class PutveinBioDatabase(BioDatabase):
from bob.db.putvein.query import Database as LowLevelDatabase
self.__db = LowLevelDatabase()
self.low_level_group_names = ('train', 'dev', 'eval')
self.high_level_group_names = ('world', 'dev', 'eval')
def groups(self):
return self.convert_names_to_highlevel(self.__db.groups(),
self.low_level_group_names, self.high_level_group_names)
def __protocol_split__(self, prot_name):
"""
Overrides the "high level" database names (see the list abowe) to the
......@@ -155,6 +163,8 @@ class PutveinBioDatabase(BioDatabase):
"""
kind, prot = self.__protocol_split__(protocol)
groups = self.convert_names_to_lowlevel(groups, self.low_level_group_names, self.high_level_group_names)
return self.__db.model_ids(protocol=prot,
groups=groups,
kinds=kind)
......@@ -163,6 +173,9 @@ class PutveinBioDatabase(BioDatabase):
def objects(self, protocol=None, groups=None, purposes=None, model_ids=None, kinds=None, **kwargs):
kind, prot = self.__protocol_split__(protocol)
groups = self.convert_names_to_lowlevel(groups, self.low_level_group_names, self.high_level_group_names)
retval = self.__db.objects(protocol=prot,
groups=groups,
purposes=purposes,
......
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