Refactor _DOCUMENTCLASS variable to inline pgf.documentclass access

MiniX16 · MiniX16 · commit 0859eff37a21 · 2025-05-22T22:23:59.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']}",
@@ -836,7 +831,8 @@ def print_pdf(self, fname_or_fh, *, metadata=None, **kwargs):
             self.print_pgf(tmppath / "figure.pgf", **kwargs)
             (tmppath / "figure.tex").write_text(
                 "\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}"
                     % (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),
