Skip to content

Commit be8fc1f

Browse files
authored
fix: linting
1 parent 70b7691 commit be8fc1f

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

the_well/benchmark/trainer/training.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -314,7 +314,11 @@ def validation_loop(
314314
loss_dict = {}
315315
time_logs = {}
316316
count = 0
317-
denom = len(dataloader) if full else min(self.short_validation_length, len(dataloader))
317+
denom = (
318+
len(dataloader)
319+
if full
320+
else min(self.short_validation_length, len(dataloader))
321+
)
318322
with torch.autocast(
319323
self.device.type, enabled=self.enable_amp, dtype=self.amp_type
320324
):

0 commit comments

Comments
 (0)