Fix plot_csv.py

roc-streaming · Feb 20, 2024 · e86ccda · e86ccda
1 parent 1f21e84
commit e86ccda
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/scripts/plot/plot_csv.py b/scripts/plot/plot_csv.py
@@ -17,6 +17,9 @@
 
 data = numpy.genfromtxt(text, dtype=float, delimiter=',')
 
+if len(data.shape) == 1:
+    data = data[:,numpy.newaxis]
+
 if not columns:
     columns = [None] * data.shape[1]