ui.display not working with plotly express object

[ankit51]

Describe the Bug

Logging a plotly express object using ui.display throws an error

Expected behavior

It should log the plot to the Panels in my experiment

Where is the issue?

• [ x] Comet Python SDK
• [ ] Comet UI
• [ ] Third Party Integrations (Huggingface, TensorboardX, Pytorch Lighting etc)

To Reproduce

Steps to reproduce the behavior:

1. Go to '...'

2. Click on '....'

3. Scroll down to '....'

4. See error

fig = px.histogram(train, x="SalePrice", marginal="rug",
                   hover_data=train.columns)
ui.display(fig)

Stack Trace

If possible please include the full stack trace of your issue here

# Paste stack trace here

Comet Debug Log

If possible, please follow the instructions here to run Comet in debug mode and attach the resulting log file.

Screenshots or GIFs

If applicable, add screenshots/gifs to help explain your problem.

Link to Comet Project/Experiment

If applicable, please provide a link to your Comet Project or Experiment. https://www.comet.com/ankit51/px/80d13e6671ef4c1f84c70358af0a7762?experiment-tab=chart&showOutliers=true&smoothing=0&transformY=smoothing&xAxis=step

Additional context

Add any other context about the problem here.