Import numpy assert_raises for doctest

jni · Sep 30, 2015 · 609862c · 609862c
1 parent 207bc5f
commit 609862c
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/gala/classify.py b/gala/classify.py
@@ -9,6 +9,7 @@
 # libraries
 import h5py
 import numpy as np
+from numpy.testing import assert_raises
 np.seterr(divide='ignore')
 
 from sklearn.svm import SVC