From 522361d3182ec599a3314148fc296fbb38fc349d Mon Sep 17 00:00:00 2001
From: Amir Mohammadi <amir.mohammadi@idiap.ch>
Date: Wed, 5 Oct 2016 15:43:25 +0200
Subject: [PATCH] [bob.db.replay] Fix model_ids

---
 bob/bio/face/database/replay.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/bob/bio/face/database/replay.py b/bob/bio/face/database/replay.py
index c85baba9..2fe1815d 100644
--- a/bob/bio/face/database/replay.py
+++ b/bob/bio/face/database/replay.py
@@ -68,7 +68,7 @@ class ReplayBioDatabase(BioDatabase):
 
     def model_ids_with_protocol(self, groups=None, protocol=None, **kwargs):
         # since the low-level API does not support verification straight-forward-ly, we improvise.
-        files = self.objects(groups=groups, protocol=protocol)
+        files = self.objects(groups=groups, protocol=protocol, purposes='enroll', **kwargs)
         return sorted(set(f.client_id for f in files))
 
     def objects(self, groups=None, protocol=None, purposes=None, model_ids=None, **kwargs):
@@ -104,7 +104,8 @@ class ReplayBioDatabase(BioDatabase):
         elif '-spoof' in protocol:
             protocol = protocol.replace('-spoof', '')
             # you need to replace probe with attack and real for the spoof protocols.
-            # I am adding the real here also to create positives scores also.
+            # You can add the real here also to create positives scores also
+            # but usually you get these scores when you run the licit protocol
             if 'probe' in purposes:
                 purposes.remove('probe')
                 purposes.append('attack')
-- 
GitLab