Skip to content
GitLab
Projects
Groups
Snippets
/
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
Menu
Open sidebar
bob
bob.pad.face
Commits
9ea5856e
Commit
9ea5856e
authored
Dec 04, 2019
by
Guillaume HEUSCH
Browse files
[databases] proper handling when either groups or purposes are not given for NRSU
parent
c94f2942
Changes
1
Hide whitespace changes
Inline
Side-by-side
bob/pad/face/database/brsu.py
View file @
9ea5856e
...
...
@@ -140,7 +140,8 @@ class BRSUPadDatabase(PadDatabase):
samples : :py:class:`BRSUPadFile`
A list of BRSUPadFile objects.
"""
if
groups
is
not
None
:
lowlevel_purposes
=
None
if
groups
is
not
None
and
purposes
is
not
None
:
# for training
lowlevel_purposes
=
[]
...
...
@@ -155,6 +156,13 @@ class BRSUPadDatabase(PadDatabase):
if
'test'
in
groups
and
'attack'
in
purposes
:
lowlevel_purposes
.
append
(
'attack'
)
if
groups
is
None
and
purposes
is
not
None
:
lowlevel_purposes
=
[]
if
'real'
in
purposes
:
lowlevel_purposes
.
append
(
'real'
)
if
'attack'
in
purposes
:
lowlevel_purposes
.
append
(
'attack'
)
samples
=
self
.
db
.
objects
(
groups
=
groups
,
purposes
=
lowlevel_purposes
,
**
kwargs
)
samples
=
[
BRSUPadFile
(
s
)
for
s
in
samples
]
return
samples
...
...
Write
Preview
Supports
Markdown
0%
Try again
or
attach a new file
.
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment