old_contents: """Uncertainty Baselines.
  
  See more details in the
  [`README.md`](https://github.com/google/uncertainty-baselines).
  """
  
  import os
  import sys
  
  from setuptools import find_packages
  from setuptools import setup
  
  # To enable importing version.py directly, we add its path to sys.path.
  version_path = os.path.join(os.path.dirname(__file__), 'uncertainty_baselines')
  sys.path.append(version_path)
  from version import __version__  # pylint: disable=g-import-not-at-top
  
  setup(
      name='uncertainty_baselines',
      version=__version__,
      description='Uncertainty Baselines',
      author='Uncertainty Baselines Team',
      author_email='znado@google.com',
      url='http://github.com/google/uncertainty-baselines',
      license='Apache 2.0',
      packages=find_packages(),
      install_requires=[
          'absl-py>=0.8.1',
          'numpy>=1.7',
          'tf-nightly',
          'tensorboard',
          'tensorflow-datasets>=1.3.0',
          # https://github.com/tensorflow/tensorflow/issues/44146 is reproducible
          # with tf-nightly
          'gast==0.3.3'
      ],
      extras_require={
          'experimental': [
              'robustness_metrics @ git+https://github.com/google-research/robustness_metrics.git#egg=robustness_metrics',
              'uncertainty_metrics[tensorflow]',
          ],
          'models': [
              'tf-models-nightly',  # Needed for BERT, depends on tf-nightly.
              'tensorflow-probability',
              'edward2 @ git+https://github.com/google/edward2.git#egg=edward2[tf-nightly]',
          ],
          'tests': ['pylint>=1.9.0'],
      },
      classifiers=[
          'Development Status :: 3 - Alpha',
          'Intended Audience :: Developers',
          'Intended Audience :: Science/Research',
          'License :: OSI Approved :: Apache Software License',
          'Topic :: Scientific/Engineering :: Artificial Intelligence',
      ],
      keywords='probabilistic programming tensorflow machine learning',
  )
  
new_contents: """Uncertainty Baselines.
  
  See more details in the
  [`README.md`](https://github.com/google/uncertainty-baselines).
  """
  
  import os
  import sys
  
  from setuptools import find_packages
  from setuptools import setup
  
  # To enable importing version.py directly, we add its path to sys.path.
  version_path = os.path.join(os.path.dirname(__file__), 'uncertainty_baselines')
  sys.path.append(version_path)
  from version import __version__  # pylint: disable=g-import-not-at-top
  
  setup(
      name='uncertainty_baselines',
      version=__version__,
      description='Uncertainty Baselines',
      author='Uncertainty Baselines Team',
      author_email='znado@google.com',
      url='http://github.com/google/uncertainty-baselines',
      license='Apache 2.0',
      packages=find_packages(),
      install_requires=[
          'absl-py>=0.8.1',
          'numpy>=1.7',
          'tf-nightly',
          'tensorboard',
          'tensorflow-datasets>=1.3.0',
          # https://github.com/tensorflow/tensorflow/issues/44146 is reproducible
          # with tf-nightly
          'gast==0.3.3',
          'astunparse',
          'opt_einsum',
          'astunparse',
          'flatbuffers',
          'zipp',
          'urllib3',
          'chardet',
          'idna'
      ],
      extras_require={
          'experimental': [
              'robustness_metrics @ git+https://github.com/google-research/robustness_metrics.git#egg=robustness_metrics',
              'uncertainty_metrics[tensorflow]',
          ],
          'models': [
              'tf-models-nightly',  # Needed for BERT, depends on tf-nightly.
              'tensorflow-probability',
              'edward2 @ git+https://github.com/google/edward2.git#egg=edward2[tf-nightly]',
              'pandas',
              'scipy'
          ],
          'tests': ['pylint>=1.9.0'],
      },
      classifiers=[
          'Development Status :: 3 - Alpha',
          'Intended Audience :: Developers',
          'Intended Audience :: Science/Research',
          'License :: OSI Approved :: Apache Software License',
          'Topic :: Scientific/Engineering :: Artificial Intelligence',
      ],
      keywords='probabilistic programming tensorflow machine learning',
  )
  
subject: update depd