Fixes error in handling export outputs.

yeatmanlab · Sep 27, 2023 · 9ad67a2 · 9ad67a2
1 parent 1037bc1
commit 9ad67a2
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/AFQ/api/bundle_dict.py b/AFQ/api/bundle_dict.py
@@ -962,7 +962,7 @@ def load_templates(self):
 
         self.templates["Callosum_midsag"] = self.templates["mid-saggital"]
 
-        for bundle_name in PEDIATRIC_BUNDLES:
+        for bundle_name in prob_map_order:
             self.templates[bundle_name + "_prob_map"] = nib.Nifti1Image(
                 prob_map_data[
                     ...,

diff --git a/examples/tutorial_examples/plot_baby_afq.py b/examples/tutorial_examples/plot_baby_afq.py
@@ -77,8 +77,8 @@
 ##########################################################################
 # Visualizing bundles:
 # --------------------
-
-plotly.io.show(myafq.export("all_bundles_figure").values()[0][0])
+viz = myafq.export("all_bundles_figure")
+plotly.io.show(viz["01"])
 
 ##########################################################################
 # References: