diff --git a/src/HydroMod.py b/src/HydroMod.py index 2d89d30..4901793 100644 --- 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):