diff --git a/libact/base/dataset.py b/libact/base/dataset.py index 8684116d..acd63430 100644 --- a/libact/base/dataset.py +++ b/libact/base/dataset.py @@ -34,12 +34,14 @@ class Dataset(object): """ def __init__(self, X=None, y=None): - if X is None: X = np.array([]) + if X is None: + X = np.array([]) elif not isinstance(X, sp.csr_matrix): X = np.array(X) - if y is None: y = [] - y = np.array(y) + if y is None: + y = [] + y = np.array(y, dtype=object) self._X = X self._y = y diff --git a/libact/query_strategies/_hintsvm.pyx b/libact/query_strategies/_hintsvm.pyx index dd6ae0b6..e7208247 100644 --- a/libact/query_strategies/_hintsvm.pyx +++ b/libact/query_strategies/_hintsvm.pyx @@ -3,7 +3,7 @@ import numpy as np cimport numpy as np from libc.stdlib cimport free -cimport _hintsvm +from . cimport _hintsvm cdef extern from *: ctypedef struct svm_parameter: diff --git a/requirements.txt b/requirements.txt index 85c09f3b..37c796a0 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,4 +1,4 @@ -setuptools +setuptools<60.0 numpy scipy scikit-learn>=0.24