google colaboratory でmatplotlib 3d surface のアスペクト比を変更したい

Question

前提

google colaboratoryで matplotlib 3d surface のアスペクト比を変更したいのですが、Jupiter notebookで上手くいくコードがなぜかエラーを吐いてしまいます。ご教示いただけないでしょうか。よろしくお願い致します。

実現したいこと

google colaboratoryでmatplotlib 3d surface のアスペクト比を変更したい

発生している問題・エラーメッセージ

matplotlib 3d のアスペクト比を変更するためset_box_aspect()を使用してみた。
結果、Jupiter notebookでは上手くアスペクト比を変更できたのに対し、google colaboratoryではエラーとなった。
以下、google colaboratoryでset_box_aspect()を使用した際のエラーメッセージ

'Axes3DSubplot' object has no attribute 'set_box_aspect'

該当のソースコード

!pip install matplotlib==3.3
from mpl_toolkits.mplot3d import Axes3D 
import matplotlib.pyplot as plt 
from matplotlib import cm
from matplotlib.colors import LightSource
import numpy as np

x = np.arange(100)
y = np.arange(200)
x, y = np.meshgrid(x, y)
z = np.sin(x + y)
xmin,xmax = 0,100
ymin,ymax = 0,200
zmin,zmax = int(np.round(np.min(z))),int(np.round(np.max(z)))

fig = plt.figure()
ax = fig.add_subplot(projection='3d')
ax.plot_surface(x, y, z)
ax.view_init(elev=90, azim=90)
ax.set_box_aspect((xmax-xmin,ymax-ymin,zmax-zmin))
ax.set_xlim3d(xmin, xmax)
ax.set_ylim3d(ymin, ymax)
ax.set_zlim3d(zmin, zmax)

試したこと

Jupiter notebookで該当のソースコードと全く同じコードを入力し、実行してみたところ上手くアスペクト比が変更された。（google colaboratoryのようなエラーが発生しなかった。）

補足情報（FW/ツールのバージョンなど）

ここにより詳細な情報を記載してください。

Answer 1

原因はわかりませんが、ランタイムを更新して再度実行すると上手く行きました。