train_autoencoder.ipynb is broken on Colab

[mindplay-dk]

tried to run this on Colab:

https://github.com/magenta/ddsp/blob/main/ddsp/colab/demos/train_autoencoder.ipynb

it doesn't work.

  Preparing metadata (setup.py) ... done
  Preparing metadata (setup.py) ... done
     ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 54.6/54.6 kB 7.8 MB/s eta 0:00:00
     ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 307.7/307.7 kB 36.7 MB/s eta 0:00:00
     ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 1.7/1.7 MB 69.3 MB/s eta 0:00:00
     ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 311.2/311.2 kB 30.3 MB/s eta 0:00:00
     ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 1.6/1.6 MB 86.6 MB/s eta 0:00:00
  Building wheel for avro-python3 (setup.py) ... done
  Building wheel for crepe (setup.py) ... done
  error: subprocess-exited-with-error
  
  × python setup.py bdist_wheel did not run successfully.
  │ exit code: 1
  ╰─> See above for output.
  
  note: This error originates from a subprocess, and is likely not a problem with pip.
  Building wheel for numba (setup.py) ... error
  ERROR: Failed building wheel for numba
  Building wheel for cloudml-hypertune (setup.py) ... done
  Building wheel for mir-eval (setup.py) ... done
  Building wheel for crcmod (setup.py) ... done
  Building wheel for dill (setup.py) ... done
  Building wheel for hdfs (setup.py) ... done
  Building wheel for intervaltree (setup.py) ... done
  error: subprocess-exited-with-error
  
  × python setup.py bdist_wheel did not run successfully.
  │ exit code: 1
  ╰─> See above for output.
  
  note: This error originates from a subprocess, and is likely not a problem with pip.
  Building wheel for llvmlite (setup.py) ... error
  ERROR: Failed building wheel for llvmlite
  Building wheel for pretty-midi (setup.py) ... done
  Building wheel for resampy (setup.py) ... done
  Building wheel for pyjsparser (setup.py) ... done
  Building wheel for docopt (setup.py) ... done
ERROR: Could not build wheels for numba, llvmlite, which is required to install pyproject.toml-based projects

I'm seeing this warning in the output as well:

 WARNING: Generating metadata for package tfds-nightly produced metadata for project name tensorflow-datasets. Fix your #egg=tfds-nightly fragments.

Might be related to #514