fix(tasks): use parameterized state color dictionary

Signed-off-by: Cameron Smith <cameron.ray.smith@gmail.com>
pinellolab · Sep 6, 2024 · e5b9719 · e5b9719
1 parent 5aab2a5
commit e5b9719
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/pyrovelocity/tasks/summarize.py b/src/pyrovelocity/tasks/summarize.py
@@ -334,6 +334,9 @@ def summarize_dataset(
             selected_genes=selected_genes,
             vector_field_basis=vector_field_basis,
             cell_state=cell_state,
+            state_color_dict=LARRY_CELL_TYPE_COLORS
+            if "larry" in data_model
+            else None,
             report_file_path=gene_selection_summary_plot,
             figure_file_path=f"{gene_selection_summary_plot}.dill.zst",
         )