diff --git a/src/dvclive/plots/image.py b/src/dvclive/plots/image.py index 5cc65362..94f4b2a1 100644 --- a/src/dvclive/plots/image.py +++ b/src/dvclive/plots/image.py @@ -37,7 +37,7 @@ def dump(self, val, **kwargs) -> None: # noqa: ARG002 elif isinstance_without_import(val, "matplotlib.figure", "Figure"): import matplotlib.pyplot as plt - plt.savefig(self.output_path) + val.savefig(self.output_path) plt.close(val) elif isinstance_without_import(val, "PIL.Image", "Image"): val.save(self.output_path)