From 6990290951bcd7ae70bc96a39881d7f636408f42 Mon Sep 17 00:00:00 2001 From: Jon Bolin Date: Mon, 9 Oct 2023 22:36:20 +0000 Subject: [PATCH] yapf --- test/spmd/test_xla_spmd_python_api_interaction.py | 1 + 1 file changed, 1 insertion(+) diff --git a/test/spmd/test_xla_spmd_python_api_interaction.py b/test/spmd/test_xla_spmd_python_api_interaction.py index edb51ac9d91..8f9c319ac54 100644 --- a/test/spmd/test_xla_spmd_python_api_interaction.py +++ b/test/spmd/test_xla_spmd_python_api_interaction.py @@ -135,6 +135,7 @@ def test_xla_autocast_api(self): class BasicDistributedTest(test_xla_sharding_base.XlaShardingTest): + @classmethod def setUpClass(cls): xr.use_spmd()