Benchmark for Revereb is breaking

Upgrading Reverb from dm-reverb-nightly-0.1.0.dev20200616 to dm-reverb-nightly-0.1.0.dev20200708 breaks benchmark2.

2020-07-12 05:22:33.299418: W tensorflow/core/framework/op_kernel.cc:1760] OP_REQUIRES failed at client.cc:234 : Invalid argument: Trying to access resource using the wrong type. Expected localhost got N8deepmind6reverb12_GLOBAL__N_114ClientResourceE

Traceback (most recent call last):
  File "benchmark.py", line 156, in <module>
    equality_check = None)
  File "/usr/local/lib/python3.7/site-packages/perfplot/main.py", line 297, in save
    out = bench(*args, **kwargs)
  File "/usr/local/lib/python3.7/site-packages/perfplot/main.py", line 202, in bench
    t, total_time = _b(data, kernel, repeat, timer, is_ns_timer, resolution)
  File "/usr/local/lib/python3.7/site-packages/perfplot/main.py", line 232, in _b
    stmt=lambda: kernel(data), repeat=repeat, number=number, timer=timer
  File "/usr/local/lib/python3.7/timeit.py", line 238, in repeat
    return Timer(stmt, setup, timer, globals).repeat(repeat, number)
  File "/usr/local/lib/python3.7/timeit.py", line 205, in repeat
    t = self.timeit(number)
  File "/usr/local/lib/python3.7/timeit.py", line 177, in timeit
    timing = self.inner(it, self.timer)
  File "<timeit-src>", line 6, in inner
  File "/usr/local/lib/python3.7/site-packages/perfplot/main.py", line 232, in <lambda>
    stmt=lambda: kernel(data), repeat=repeat, number=number, timer=timer
  File "benchmark.py", line 107, in add
    tf.constant([1.0],dtype=tf.float64))
  File "/usr/local/lib/python3.7/site-packages/reverb/tf_client.py", line 114, in insert
    self._handle, data, tables, priorities, name=scope)
  File "<string>", line 336, in reverb_client_insert
  File "/usr/local/lib/python3.7/site-packages/tensorflow/python/framework/ops.py", line 6851, in raise_from_not_ok_status
    six.raise_from(core._status_to_exception(e.code, message), None)
  File "<string>", line 3, in raise_from
tensorflow.python.framework.errors_impl.InvalidArgumentError: Trying to access resource using the wrong type. Expected localhost got N8deepmind6reverb12_GLOBAL__N_114ClientResourceE [Op:ReverbClientInsert] name: reverb_insert/
[reverb/cc/platform/default/server.cc:64] Shutting down replay server