From 07acb394e9218b68e7f0eef9f308e8c9e27e9c1c Mon Sep 17 00:00:00 2001 From: Will Cromar Date: Fri, 17 May 2024 19:49:01 +0000 Subject: [PATCH] fix test --- test/test_devices.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/test/test_devices.py b/test/test_devices.py index 4588940fb80..259f0046623 100644 --- a/test/test_devices.py +++ b/test/test_devices.py @@ -47,7 +47,7 @@ def test_step(self): with xla.step(): torch.ones((3, 3), device=xla.device()) - self.assertEqual(met.counter_value('MarkStep'), 1) + self.assertEqual(met.counter_value('MarkStep'), 2) def test_step_exception(self): with self.assertRaisesRegex(RuntimeError, 'Expected error'): @@ -55,7 +55,7 @@ def test_step_exception(self): torch.ones((3, 3), device=xla.device()) raise RuntimeError('Expected error') - self.assertEqual(met.counter_value('MarkStep'), 1) + self.assertEqual(met.counter_value('MarkStep'), 2) # Should roughly match example given in README def test_trivial_model(self):