Merge pull request #27 from MG-ng/patch-7

fn -> file_name
marinlauber · Jan 17, 2024 · 2fd978d · 2fd978d
2 parents e56f8c7 + e942cac
commit 2fd978d
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/HydroMod.py b/src/HydroMod.py
@@ -184,7 +184,7 @@ def _get_RmV(self, vb):
     #         a[i] = self.interp_a[i](fn)
     #     return a
 
-    def show_resistance(self, vb):
+    def show_resistance(self, vb, file_name="None"):
         resV, resR = np.empty_like(vb), np.empty_like(vb)
         for i, v in enumerate(vb):
             self.vb = v * 0.5144
@@ -200,8 +200,8 @@ def show_resistance(self, vb):
         plt.ylabel(r"$R$ (N)")
         plt.legend()
         plt.tight_layout()
-        if fn!="None":
-            plt.savefig(fn, dpi=300)
+        if file_name != "None":
+            plt.savefig(file_name, dpi=300)
         plt.show()
 
     def _test_gz(self):