go.images moved around dcc.graph in combination with plotly.make_subplots()

First off, I have also addressed this in the plotly community website, but I feel like this is the place it should be posted.

EDIT: I realize it might be better suited in the dash github. Let me know if I need to move it.

There is a bug in plotly when using dcc.Graph in combination with a subplot figure consisting of go.Images the go.Images get moved around after the user changes the number of rows of plotly.make_subplots().
For example, when I first choose to plot 5 images beneath each other, the plot will look fine, plotting five images closely together (chosen first). However, when the user subsequently chooses to plot 2 points, the images will be plotted in the same space the five images were. Similarly, when the user first chooses to plot 2 images the image looks fine. Subsequently, when the user plots 5 images, they all get messed up.

This does not happen when plotting, for example, scatter plots. Moreover, adding subtitles to the plot shows that the subtitles do not get moved around either. Using fig.show() on the figures also gives a good figure.

You can use the code below to get a minimal working example.
Moreover, using fig.show() also shows the images in their correct spaces. I believe this is a problem with dcc.graph in combination with subplots using go.Image.

from dash import Dash, dcc, html
from dash.dependencies import Input, Output
import plotly.graph_objects as go
import dash.exceptions
import numpy as np
from plotly.subplots import make_subplots
import plotly.express as px

def create_fig(n_rows):
    fig = make_subplots(n_rows, 1)

    for i in range(n_rows):
        fig.add_trace(go.Image(z=np.random.randint(0, 255, size=(60, 60, 3))), i+1, 1)
        #fig.add_trace(go.Scatter(x=[1, 2], y=[2, 4]), i + 1, 1)

    fig.update_layout(autosize=True,
                      width=100,
                      height=n_rows*100,
                      margin=dict(l=0, r=0, b=0, t=0)
                      )
    print(fig.layout)
    return fig


app = Dash(__name__)
app.layout = html.Div(
    [dcc.Graph(id='graph', style={'overflow':'scroll'}),
     dcc.Input(value=0, id='input', type='number')])


@app.callback(Output('graph', 'figure'), Input('input', 'value'))
def create_graph(n_rows):
    if n_rows == 0:
        raise dash.exceptions.PreventUpdate("Prevent update")
    else:
        if n_rows is not None:
            return create_fig(int(n_rows))


if __name__ == '__main__':
    app.run_server(debug=True)

Bắt đầu bằng cách chạy ví dụ Dash tối thiểu, tập trung vào callback của dcc.Graph, create_fig, make_subplots và các trace go.Image, đồng thời thay đổi đầu vào giữa hai và năm hàng. So sánh figure đã cập nhật với fig.show() và các trường hợp scatter và subtitle. Được xem là hoàn tất khi các hình ảnh vẫn nằm trong đúng không gian subplot của chúng sau khi số hàng thay đổi.