From 24304a85f49938cfa6d283687de38e1817c64259 Mon Sep 17 00:00:00 2001 From: Graph Learning Team Date: Wed, 6 May 2026 07:18:54 -0700 Subject: [PATCH] Reset Keras global mixed precision policy to float32 in setUp of orchestration_test.py. PiperOrigin-RevId: 911321484 --- tensorflow_gnn/runner/orchestration_test.py | 1 + 1 file changed, 1 insertion(+) diff --git a/tensorflow_gnn/runner/orchestration_test.py b/tensorflow_gnn/runner/orchestration_test.py index e0162efc..90df3d09 100644 --- a/tensorflow_gnn/runner/orchestration_test.py +++ b/tensorflow_gnn/runner/orchestration_test.py @@ -281,6 +281,7 @@ class OrchestrationTests(tf.test.TestCase, parameterized.TestCase): def setUp(self): super().setUp() tfgnn.enable_graph_tensor_validation_at_runtime() + tf.keras.mixed_precision.set_global_policy("float32") @parameterized.named_parameters([ dict(