Skip to content

Commit

Permalink
fix linter
Browse files Browse the repository at this point in the history
  • Loading branch information
vanbasten23 committed Jan 17, 2024
1 parent 1ad337d commit 878f726
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion test/pjrt/test_torchrun.py
Original file line number Diff line number Diff line change
Expand Up @@ -15,7 +15,7 @@ def setUp(self):

def tearDown(self) -> None:
dist.destroy_process_group()

def test_addressable_device_count(self):
devices_per_process = xr.addressable_device_count()
self.assertEqual(devices_per_process, 1)
Expand Down

0 comments on commit 878f726

Please sign in to comment.