Refactor _DOCUMENTCLASS variable to inline pgf.documentclass access

MiniX16 · MiniX16 · commit 1bf0be7f6244 · 2025-05-23T15:21:11.000+02:00
diff --git a/lib/matplotlib/backends/backend_pgf.py b/lib/matplotlib/backends/backend_pgf.py
@@ -29,11 +29,6 @@
 _log = logging.getLogger(__name__)
 
 
-_DOCUMENTCLASS = (
-    rf"\documentclass{{{mpl.rcParams.get('pgf.documentclass', 'article')}}}"
-)
-
-
 # Note: When formatting floating point values, it is important to use the
 # %f/{:f} format rather than %s/{} to avoid triggering scientific notation,
 # which is not recognized by TeX.
@@ -207,7 +202,7 @@ class LatexManager:
     @staticmethod
     def _build_latex_header():
         latex_header = [
-            _DOCUMENTCLASS,
+            rf"\documentclass{{{mpl.rcParams.get('pgf.documentclass', 'article')}}}",
             # Include TeX program name as a comment for cache invalidation.
             # TeX does not allow this to be the first line.
             rf"% !TeX program = {mpl.rcParams['pgf.texsystem']}",
@@ -833,10 +828,11 @@ def print_pdf(self, fname_or_fh, *, metadata=None, **kwargs):
         # print figure to pgf and compile it with latex
         with TemporaryDirectory() as tmpdir:
             tmppath = pathlib.Path(tmpdir)
+            docclass = mpl.rcParams.get("pgf.documentclass", "article")
             self.print_pgf(tmppath / "figure.pgf", **kwargs)
             (tmppath / "figure.tex").write_text(
                 "\n".join([
-                    _DOCUMENTCLASS,
+                    rf"\documentclass{{{docclass}}}"
                     r"\usepackage[pdfinfo={%s}]{hyperref}" % pdfinfo,
                     r"\usepackage[papersize={%fin,%fin}, margin=0in]{geometry}"
                     % (w, h),
@@ -933,7 +929,7 @@ def _write_header(self, width_inches, height_inches):
         pdfinfo = ','.join(
             _metadata_to_str(k, v) for k, v in self._info_dict.items())
         latex_header = "\n".join([
-            _DOCUMENTCLASS,
+            rf"\documentclass{{{mpl.rcParams.get('pgf.documentclass', 'article')}}}",
             r"\usepackage[pdfinfo={%s}]{hyperref}" % pdfinfo,
             r"\usepackage[papersize={%fin,%fin}, margin=0in]{geometry}"
             % (width_inches, height_inches),
