Fix call to make_dataset() to use server_address.
Fixes
"# adder above so we'll tell the dataset function that so that it knows the type\n",
"# of data that's coming out.\n",
"dataset = datasets.make_dataset(\n",
- " client=reverb.TFClient(server_address=replay_server_address),\n",
+ " server_address=replay_server_address,\n",
" batch_size=256,\n",
" environment_spec=environment_spec,\n",
" transition_adder=True)"
