Plotly.express in Sagemaker Studio

Hi, I am curious to know if there are any known limitations to using plotly.express within an Amazon SageMaker Studio Jupyter notebook?

I am trying to create a 3D scatter plot based on below code and the process has been running for over an hour w/o error.

import plotly.express as px
from sklearn.decomposition import PCA 

pca=PCA(n_components=3)
X3D=pca.fit_transform(X)

X3D=pd.DataFrame(X3D,columns=['pca1','pca2','pca3'])
X3D['Class']=df['Class']

# We can change the frac value to reduce the points plotted 
samp=X3D.sample(frac=1)


fig = px.scatter_3d(samp, x='pca1', y='pca2', z='pca3',
              color='Class')
fig.show()