Commit 8fef6f1c authored by Laurent EL SHAFEY's avatar Laurent EL SHAFEY
Browse files

Fix OSX issue when using labels (force list usage instead of set)

parent 5840e121
......@@ -9,7 +9,7 @@ from setuptools import setup, find_packages
setup(
name='xbob.db.mnist',
version='1.0.0e',
version='1.0.0f',
description='MNIST Database Access API for Bob',
url='http://github.com/bioidiap/xbob.db.mnist',
license='GPLv3',
......
......@@ -194,7 +194,7 @@ class Database():
labels = numpy.ndarray(shape=(0,), dtype=numpy.uint8)
# List of indices for which the labels are in the list of requested labels
indices = numpy.where(numpy.in1d(labels, vlabels))[0]
indices = numpy.where(numpy.in1d(labels, list(vlabels)))[0]
images = images[indices,:]
labels = labels[indices]
......
Supports Markdown
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