test: improve speed for AnnData with lots of metadata

Teichlab · Sep 21, 2021 · f9ff81f · f9ff81f
1 parent 13eddb8
commit f9ff81f
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/SpatialDE/_internal/tf_dataset.py b/SpatialDE/_internal/tf_dataset.py
@@ -30,11 +30,10 @@ def __init__(
 
     def __call__(self):
         for i, g in enumerate(self.genes):
-            slice = self.adata[:, i]
             if self.layer is None:
-                data = slice.X
+                data = self.adata.X[:, i]
             else:
-                data = slice.layers[self.layer]
+                data = self.adata.layers[self.layer][:, i]
             if issparse(data):
                 data = data.toarray()
             with tf.device(tf.DeviceSpec(device_type="CPU").to_string()):