fix(morph-plot): handle other general cases for NeuroMLDocument

NeuroML · Aug 21, 2024 · b7da5d9 · b7da5d9
1 parent 89f19ff
commit b7da5d9
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/pyneuroml/plot/PlotMorphology.py b/pyneuroml/plot/PlotMorphology.py
@@ -371,6 +371,8 @@ def plot_2D(
             logger.debug(f"plottable cell model is: {plottable_nml_model.cells[0]}")
             # use title from original model document
             title = nml_model.id
+        else:
+            plottable_nml_model = nml_model
 
     if title is None:
         title = f"{plottable_nml_model.id}"

diff --git a/pyneuroml/plot/PlotMorphologyVispy.py b/pyneuroml/plot/PlotMorphologyVispy.py
@@ -555,6 +555,9 @@ def plot_interactive_3D(
             logger.debug(f"plottable cell model is: {plottable_nml_model.cells[0]}")
             # use title from original model document
             title = nml_model.id
+        # other networks
+        else:
+            plottable_nml_model = nml_model
 
     if title is None:
         title = f"{plottable_nml_model.id}"