Slightly increases the tolerance of the comparison to pass on GPU (V100 open source).
Slightly increases the tolerance of the comparison to pass on GPU (V100 open source).

tf_agents.agents.cql.cql_sac_agent_test CqlSacAgentTest.testTrainWithLagrange0

Not equal to tolerance rtol=1e-06, atol=1e-06
Mismatched value: a is different from b.
not close lhs = 3.0326085090637207
not close rhs = 3.032653
not close dif = 4.449093627911793e-05
not close tol = 4.032652999999999e-06
dtype = float32, shape = ()
Mismatched elements: 1 / 1 (100%)
Max absolute difference: 4.44909363e-05
Max relative difference: 1.4670632e-05
 x: array(3.032609, dtype=float32)
 y: array(3.032653)
PiperOrigin-RevId: 398602099
Change-Id: Icd0e4055a9ca57b728299db99b83267556eb9373

tensorflow/agents,tensorflow/agents