Jupyterlab v3 extension support

[Mariko Wakabayashi]

System information

• Have I written custom code (as opposed to using a stock example script
  provided in TensorFlow Model Analysis): N/A
• OS Platform and Distribution (e.g., Linux Ubuntu 16.04): Tested in MacOS, would like support for Linux CentOS 7
• TensorFlow Model Analysis installed from (source or binary): PyPI binary
• TensorFlow Model Analysis version (use command below): 0.27.0
• Python version: 3.6
• Jupyter Notebook version: Using jupyterlab v3.0.11
• Exact command to reproduce:
  Installing TFMA

$  pip install tensorflow_model_analysis==0.27.0
$ jupyter labextension install tensorflow_model_analysis@0.27.0
$ pip install jupyterlab_widgets
$ jupyter labextension list
JupyterLab v3.0.11
/Users/mwakabayashi/opt/anaconda3/envs/jupyterlab3/share/jupyter/labextensions
        @jupyter-widgets/jupyterlab-manager v3.0.0 enabled OK (python, jupyterlab_widgets)

Other labextensions (built into JupyterLab)
   app dir: /Users/mwakabayashi/opt/anaconda3/envs/jupyterlab3/share/jupyter/lab
        tensorflow_model_analysis v0.27.0 enabled OK

Running TFMA basic notebook

1. jupyter lab
2. Ran the TFMA basic notebook with tensorflow==2.3.0.

Describe the problem

Jupyterlab 3.0 was released in January 2021. Would it be possible to get v3 support (install TFMA as a prebuilt extension) soon? At Twitter, we'd like to migrate to Jupyterlab 3 from 2, but we can't without v3 support for TFMA.

Source code / logs

Error after running render_slicing_metrics function.

In Chrome developer console

[Mariko Wakabayashi]

@mdreves Would it be possible to get a rough ETA on this? We'd ideally like to have this support by early May.

[Mike Dreves]

I believe it's already supported with tfma>=0.29.0?

@fhuanming to confirm

[Mariko Wakabayashi]

@fhuanming I got 0.27.0 working on jupyterlab v3 by updating the package.json file's dependencies and jupyterlab attributes and rebuilding.

I think adding the sharedPackages configuration fixed the No provider for: jupyter.extensions.JupyterWidgetRegistry error.

// model-analysis/tensorflow_model_analysis/notebook/jupyter/js/package.json 
...
"dependencies": {
    "@jupyter-widgets/base": "^1.1 || ^2 || ^3 || ^4",
    "lodash": "^4.17.4"
  },
  "jupyterlab": {
    "extension": "lib/labplugin",
    "sharedPackages": {
      "@jupyter-widgets/base": {
        "bundled": false,
        "singleton": true
      }
    }
  },

Can we get this updated for 0.27.0?

[Huanming Fang]

Hi @mwakaba2,

Seems this is already updated? : https://github.com/tensorflow/model-analysis/blob/master/tensorflow_model_analysis/notebook/jupyter/js/package.json

Would you mind giving it a try of 0.28.0 ?

Thanks

[Mariko Wakabayashi]

Hi @fhuanming It's updated on the master branch, but not on branch r0.27.0.

[Huanming Fang]

Yes, it's fixed >=0.28.0, not in 0.27.0. Do you have any issue to upgrade to 0.28.0 ?

[Mariko Wakabayashi]

I see, no technical issue. We depend on another team to get it upgraded, so I just wanted to see if it worked for 0.26.0 and 0.27.0.

[Mariko Wakabayashi]

thanks for the help!