hotfix with pandas 1.0

lanagarmire · Feb 19, 2020 · a3e1b10 · a3e1b10
1 parent c19bfff
commit a3e1b10
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/deepimpute/multinet.py b/deepimpute/multinet.py
@@ -345,9 +345,9 @@ def setPredictors(self, covariance_matrix, ntop=5):
                           .drop(np.intersect1d(targets, covariance_matrix.columns), axis=1)
                           )
             sorted_idx = np.argsort(-subMatrix.values, axis=1)
-            predictors = subMatrix.columns[sorted_idx[:,:ntop]].values.flatten()
+            predictors = subMatrix.columns[sorted_idx[:,:ntop].flatten()]
 
-            self.predictors.append(np.unique(predictors))
+            self.predictors.append(predictors.unique())
 
             print("Net {}: {} predictors, {} targets"
                   .format(i,len(np.unique(predictors)),len(targets)))