Remote tf.compat.v2 import (not needed).

PiperOrigin-RevId: 309683489
Change-Id: I0f01c97330441dcf34ab4875fa4dd21d7de34760

Author: aslanides

bsuite/baselines/tf/actor_critic/agent.py

@@ -30,7 +30,7 @@
 import dm_env
 from dm_env import specs
 import sonnet as snt
-import tensorflow.compat.v2 as tf
+import tensorflow as tf
 import tensorflow_probability as tfp
 import tree
 import trfl

bsuite/baselines/tf/actor_critic_rnn/agent.py

@@ -34,7 +34,7 @@
 import dm_env
 from dm_env import specs
 import sonnet as snt
-import tensorflow.compat.v2 as tf
+import tensorflow as tf
 import tree
 import trfl
 

bsuite/baselines/tf/boot_dqn/agent.py

@@ -44,7 +44,7 @@
 from dm_env import specs
 import numpy as np
 import sonnet as snt
-import tensorflow.compat.v2 as tf
+import tensorflow as tf
 import tree