From ee4d6feece276960ba63bc31a16ee092dcdc3a75 Mon Sep 17 00:00:00 2001
From: Nimrod Gileadi <nimrod@google.com>
Date: Thu, 5 Dec 2024 05:58:51 -0800
Subject: [PATCH] Reduce required accuracy in ranking_agent_test.

PiperOrigin-RevId: 703085807
Change-Id: Iefd35170f48a9a8c6ce8f5b44d5e2dc32f6b3f24
---
 tf_agents/bandits/agents/ranking_agent_test.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/tf_agents/bandits/agents/ranking_agent_test.py b/tf_agents/bandits/agents/ranking_agent_test.py
index 421d59d7f..435d77d40 100644
--- a/tf_agents/bandits/agents/ranking_agent_test.py
+++ b/tf_agents/bandits/agents/ranking_agent_test.py
@@ -463,7 +463,9 @@ def testPositionalBiasParams(
     agent.train(experience)
     weights = agent._construct_sample_weights(scores, observations, None)
     self.assertAllEqual(weights.shape, [batch_size, num_slots])
-    self.assertAllClose(weights[-1, 1], expected_second_weight)
+    self.assertAllClose(
+        weights[-1, 1], expected_second_weight, atol=1e-3, rtol=1e-3
+    )
 
 
 if __name__ == '__main__':