fixed deprecated warning in database.py

sdevenes · Sep 20, 2020 · b8aab31 · b8aab31
1 parent d6480b7
commit b8aab31
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/scripts/database.py b/scripts/database.py
@@ -113,4 +113,4 @@ def get(protocol, subset, classes=CLASSES, variables=VARIABLES, setname='csh101'
     retval = split_data(load(setname), subset, PROTOCOLS[protocol])
     varindex = [VARIABLES.index(k) for k in variables]
     retval = dict([(k, retval[k][:, varindex]) for k in classes])
-    return np.array([retval[k] for k in classes])
+    return np.array([retval[k] for k in classes], dtype=object)