Formatando figura Matplotlib com subtramas para Latex

Formatando figura Matplotlib com subtramas para Latex

Estou tendo problemas para fazer com que meu gráfico abranja todo o arquivo \textwidth. Aqui está um MWE equivalente ao que estou obtendo quando compilo.insira a descrição da imagem aqui

Código Matplotlib: ( set_sizeé uma função retirada deaqui, mas isso também não funciona)

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")

Código de látex:

\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}

o que estou perdendo?

informação relacionada