Use search instead of match

Project-Platypus · Sep 5, 2024 · 73d5d25 · 73d5d25
1 parent e29310e
commit 73d5d25
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/rhodium/test/plot_to_savefig.py b/rhodium/test/plot_to_savefig.py
@@ -14,11 +14,11 @@
 
     with open(sys.argv[1], "r") as f:
         for line in f:
-            m = re.match(r"import matplotlib.pyplot as ([a-zA-Z_]+)", line)
+            m = re.search(r"import matplotlib.pyplot as ([a-zA-Z_]+)", line)
             if m:
                 pyplot_alias = m.group(1)
 
-            m = re.match(pyplot_alias + r".show\(\)", line)
+            m = re.search(pyplot_alias + r".show\(\)", line)
             if m:
                 line = line[:m.start()] + pyplot_alias + ".savefig('" + str(fig_basename) + "." + str(fig_index) + ".png')" + line[m.end():]
                 fig_index += 1