Formato de figura de Matplotlib con subtramas para Latex

Question

Tengo problemas para que mi trama abarque todo el archivo \textwidth. Aquí hay un MWE equivalente de lo que obtengo cuando compilo.

Código Matplotlib: ( set_sizees una función tomada deaquí, pero eso tampoco 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}

¿Qué me estoy perdiendo?