tf_agents.policies.policy_saver.rename_spec_with_nest_paths