Fix for plotting method

openworm · Nov 9, 2023 · 768aca0 · 768aca0
1 parent c94491d
commit 768aca0
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/c302/c302_utils.py b/c302/c302_utils.py
@@ -7,6 +7,7 @@
 import matplotlib.pyplot as plt
 import numpy as np
 from pyneuroml import pynml
+from pyneuroml import plot as pyneuroml_plot
 from owmeta_core.bundle import Bundle
 
 import c302
@@ -92,7 +93,7 @@ def generate_traces_plot(config,parameter_set,xvals,yvals,info,labels,save,save_
 
     file_name = 'traces_%s%s_%s_%s.png'%(('muscles' if muscles else 'neuron'),('' if voltage else '_activity'),config,parameter_set)
 
-    pynml.generate_plot(xvals,
+    pyneuroml_plot.generate_plot(xvals,
                         yvals,
                         info,
                         labels=labels,