Fix sparse COO creation

py-why · Jun 9, 2020 · b1fec80 · b1fec80
1 parent 481fad3
commit b1fec80
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/econml/utilities.py b/econml/utilities.py
@@ -708,8 +708,9 @@ def filter_inds(coords, data, n):
         for (c, d) in l:
             results[tuple(c[i] for i in coordMap)] += d
 
-    return sp.COO(np.array([k for k in results.keys()]).T,
-                  np.array([v for v in results.values()]),
+    return sp.COO(np.array(list(results.keys())).T if results else
+                  np.empty((len(outputs), 0)),
+                  np.array(list(results.values())),
                   [arrs[indMap[c][0][0]].shape[indMap[c][0][1]] for c in outputs])
 
 

diff --git a/setup.cfg b/setup.cfg
@@ -41,7 +41,7 @@ install_requires =
     scipy != 1.4.0
     scikit-learn ~= 0.21.0
     keras
-    sparse != 0.10
+    sparse
     tensorflow == 1.*
     joblib >= 0.13.0
     numba != 0.42.1