Форматирование рисунка Matplotlib с подграфиками для Latex

Question

У меня возникли проблемы с тем, чтобы мой график охватывал весь \textwidth. Вот эквивалентный MWE того, что я получаю при компиляции.

Код Matplotlib: (set_size — это функция, взятая изздесь, но это тоже не работает)

import matplotlib.pyplot as plt
import seaborn as sns

fmri = sns.load_dataset("fmri")
fig, ax = plt.subplots(1, 4, figsize=(28,5))
# fig, ax = plt.subplots(1, 4, figsize=set_size(465, subplots=(1,4)))

for i in range (4):
    sns.lineplot(x="timepoint", y="signal", err_style="bars", ci=68, ax=ax[i], data=fmri)
plt.suptitle("Timepoint vs Signal")
plt.savefig("timepoint_vs_signal.png")

Код латекса:

\documentclass{article}
\usepackage[margin=1in]{geometry}
\usepackage{subfig,graphicx}
\usepackage[lined, boxed, ruled, commentsnumbered]{algorithm2e}
\usepackage{mathtools,array,savesym}
\usepackage{amsfonts,amsthm, amsmath}
\usepackage{lipsum}  
% these are the packages I need, so I'm including them just in case it's necessary info

\begin{document}
 
\lipsum[1]

% \showthe\textwidth % = ~469pt

\begin{figure}[ht]
\begin{center}
\includegraphics[width=\textwidth]{timepoint_vs_signal.png}
\end{center}
\end{figure}

\end{document}

Что я упускаю?