From d82f3a017b31d2d1572904ea3643b1306643e53b Mon Sep 17 00:00:00 2001 From: Nachiket Mokashi Date: Mon, 9 Dec 2024 18:05:19 +0000 Subject: [PATCH] Lint fix - add space after classname --- test/test_autocast_xla.py | 1 + 1 file changed, 1 insertion(+) diff --git a/test/test_autocast_xla.py b/test/test_autocast_xla.py index 301cd27d0b0..958c5e76620 100644 --- a/test/test_autocast_xla.py +++ b/test/test_autocast_xla.py @@ -8,6 +8,7 @@ class TestAutocastXla(unittest.TestCase): + def test_cross_entropy_loss(self): data = torch.randn(16, 10).to(torch.bfloat16).to(device) target = torch.randn(16, 10).to(torch.bfloat16).to(device)