handles non integer X matrix

hubmapconsortium · Nov 6, 2023 · 4f2d1c3 · 4f2d1c3
1 parent b545052
commit 4f2d1c3
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/containers/popv/context/main.py b/containers/popv/context/main.py
@@ -62,12 +62,16 @@ def prepare_query(
         model_path = self.find_model_dir(options["models_dir"], organ)
         reference_data = scanpy.read_h5ad(reference_data_path)
         n_samples_per_label = self.get_n_samples_per_label(reference_data, options)
-        data, var_names = self.normalize_var_names(data, options)
+        data = self.normalize_var_names(data, options)
 
         if options["query_layers_key"] == "raw":
             options["query_layers_key"] = None
             data.X = data.raw.X
 
+        if options["query_layers_key"] == "X":
+            options["query_layers_key"] = None
+            data.X = numpy.rint(data.X)
+
         data = self.add_model_genes(data, model_path, options["query_layers_key"])
         data.var_names_make_unique()
 
@@ -149,7 +153,7 @@ def getNewName(name: str):
             return lookup.get(key, name)
 
         data.var_names = t.cast(t.Any, names.map(getNewName))
-        return data, names
+        return data
 
     def load_ensemble_lookup(self, options: PopvOptions):
         with open(options["ensemble_lookup"]) as file: