Only use larger terminal width on extension doctests

Author: sethaxen

docs/make.jl

@@ -29,37 +29,34 @@ doctestfilters = [
     r"\s+\"created_at\" => .*",  # ignore timestamps in doctests
 ]
 
-# Increase the terminal width from 80 to 100 chars to avoid column truncation
-withenv("COLUMNS" => 100) do
-    makedocs(;
-        modules=[
-            InferenceObjects,
-            Base.get_extension(InferenceObjects, :InferenceObjectsMCMCDiagnosticToolsExt),
-            Base.get_extension(InferenceObjects, :InferenceObjectsPosteriorStatsExt),
-        ],
-        authors="Seth Axen <[email protected]> and contributors",
-        repo=Remotes.GitHub("arviz-devs", "InferenceObjects.jl"),
-        sitename="InferenceObjects.jl",
-        format=Documenter.HTML(;
-            prettyurls=get(ENV, "CI", "false") == "true",
-            canonical="https://arviz-devs.github.io/InferenceObjects.jl",
-            edit_link="main",
-            assets=String[],
-        ),
-        pages=[
-            "Home" => "index.md",
-            "Dataset" => "dataset.md",
-            "InferenceData" => "inference_data.md",
-            "Extensions" => [
-                "MCMCDiagnosticTools" => "extensions/mcmcdiagnostictools.md",
-                "PosteriorStats" => "extensions/posteriorstats.md",
-            ],
+makedocs(;
+    modules=[
+        InferenceObjects,
+        Base.get_extension(InferenceObjects, :InferenceObjectsMCMCDiagnosticToolsExt),
+        Base.get_extension(InferenceObjects, :InferenceObjectsPosteriorStatsExt),
+    ],
+    authors="Seth Axen <[email protected]> and contributors",
+    repo=Remotes.GitHub("arviz-devs", "InferenceObjects.jl"),
+    sitename="InferenceObjects.jl",
+    format=Documenter.HTML(;
+        prettyurls=get(ENV, "CI", "false") == "true",
+        canonical="https://arviz-devs.github.io/InferenceObjects.jl",
+        edit_link="main",
+        assets=String[],
+    ),
+    pages=[
+        "Home" => "index.md",
+        "Dataset" => "dataset.md",
+        "InferenceData" => "inference_data.md",
+        "Extensions" => [
+            "MCMCDiagnosticTools" => "extensions/mcmcdiagnostictools.md",
+            "PosteriorStats" => "extensions/posteriorstats.md",
         ],
-        doctestfilters=doctestfilters,
-        warnonly=:missing_docs,
-        plugins=[links],
-    )
-end
+    ],
+    doctestfilters=doctestfilters,
+    warnonly=:missing_docs,
+    plugins=[links],
+)
 
 # run doctests on extensions
 function get_extension(mod::Module, name::Symbol)
@@ -77,7 +74,10 @@ for extended_pkg in (MCMCDiagnosticTools, PosteriorStats)
     @info "Running doctests for extension $(extension_name)"
     mod = get_extension(InferenceObjects, extension_name)
     DocMeta.setdocmeta!(mod, :DocTestSetup, :(using $(Symbol(extended_pkg))))
-    doctest(mod; manual=false)
+    # Increase the terminal width from 80 to 100 chars to avoid column truncation
+    withenv("COLUMNS" => 100) do
+        doctest(mod; manual=false)
+    end
 end
 
 deploydocs(;