Skip to content

Commit

Permalink
format
Browse files Browse the repository at this point in the history
  • Loading branch information
will-cromar committed May 16, 2024
1 parent e69279d commit a906b0c
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 1 deletion.
2 changes: 2 additions & 0 deletions test/test_devices.py
Original file line number Diff line number Diff line change
Expand Up @@ -58,7 +58,9 @@ def test_step_exception(self):

# Should roughly match example given in README
def test_trivial_model(self):

class TrivialModel(nn.Module):

def __init__(self):
super().__init__()
self.linear = nn.Linear(10, 10)
Expand Down
1 change: 0 additions & 1 deletion torch_xla/torch_xla.py
Original file line number Diff line number Diff line change
Expand Up @@ -64,4 +64,3 @@ def step():
"""
yield
xm.mark_step()

0 comments on commit a906b0c

Please sign in to comment.