Skip to content

Commit 7706e76

Browse files
authored
🎨 Format Python code with psf/black
1 parent 93500ae commit 7706e76

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

dacapo/experiments/model.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -185,7 +185,9 @@ def __get_output_shape(
185185
The output shape is the spatial shape of the model, i.e., not accounting for channels and batch dimensions.
186186
187187
"""
188-
dummy_data = torch.zeros((1, in_channels) + input_shape, device=self.get_device())
188+
dummy_data = torch.zeros(
189+
(1, in_channels) + input_shape, device=self.get_device()
190+
)
189191
with torch.no_grad():
190192
out = self.forward(dummy_data)
191193
return out.shape[1], Coordinate(out.shape[2:])

0 commit comments

Comments
 (0)