Skip to content

Commit 0b49965

Browse files
committed
refactor: apply linting
Signed-off-by: Naren Dasan <[email protected]> Signed-off-by: Naren Dasan <[email protected]>
1 parent ff87956 commit 0b49965

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

tests/modules/hub.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -100,6 +100,7 @@ def forward(self, x):
100100

101101
# Sample Conditional Model (for testing partitioning and fallback in conditionals)
102102
class FallbackIf(torch.nn.Module):
103+
103104
def __init__(self):
104105
super(FallbackIf, self).__init__()
105106
self.relu1 = torch.nn.ReLU()
@@ -121,6 +122,7 @@ def forward(self, x):
121122
x = self.conv1(x)
122123
return x
123124

125+
124126
conditional_model = FallbackIf().eval().cuda()
125127
conditional_script_model = torch.jit.script(conditional_model)
126128
torch.jit.save(conditional_script_model, "conditional_scripted.jit.pt")

0 commit comments

Comments
 (0)