Skip to content

Commit

Permalink
style
Browse files Browse the repository at this point in the history
  • Loading branch information
ziw-liu committed Aug 9, 2023
1 parent 8fcc41b commit 6edebc2
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion viscy/light/data.py
Original file line number Diff line number Diff line change
Expand Up @@ -433,7 +433,7 @@ def _setup_predict(self, dataset_settings: dict):
)

def on_before_batch_transfer(self, batch: Sample, dataloader_idx: int) -> Sample:
predicting= False
predicting = False
if self.trainer:
if self.trainer.predicting:
predicting = True
Expand Down

0 comments on commit 6edebc2

Please sign in to comment.