Update Sonnet version to 2.0.0.

PiperOrigin-RevId: 303292840
Change-Id: I0118990892ac9c369c94de966a092f5056d1eafd

Author: tomhennigan

README.md

@@ -52,7 +52,7 @@ To get started install TensorFlow 2.0 and Sonnet 2:
 
 ```shell
 $ pip install tensorflow-gpu tensorflow-probability
-$ pip install "dm-sonnet>=2.0.0b0" --pre
+$ pip install dm-sonnet
 ```
 
 You can run the following to verify things installed correctly:

docs/index.rst

@@ -20,7 +20,7 @@ Installation
 Install Sonnet by running::
 
     $ pip install tensorflow-gpu
-    $ pip install "dm-sonnet>=2.0.0b0" --pre
+    $ pip install dm-sonnet
 
 .. toctree::
    :caption: Guides

sonnet/__init__.py

@@ -149,7 +149,7 @@
     "static_unroll",
 )
 
-__version__ = "2.0.0b0"
+__version__ = "2.0.0"
 
 #  ________________________________________
 # / Please don't use symbols in `src` they \

sonnet/src/optimizers/optimizer_tests.py

@@ -105,6 +105,9 @@ def testInconsistentDTypes(self):
 
   def testUnsuppportedStrategyError(self):
     strategy = tf.distribute.MirroredStrategy()
+    if not hasattr(strategy, "run"):
+      # TODO(tomhennigan) Remove this once we test with TF 2.2.
+      strategy.run = strategy.experimental_run_v2
     with strategy.scope():
       parameters = [tf.Variable(1.0)]
       updates = [tf.constant(0.1)]