autofixes from ruff

This commit is contained in:
AUTOMATIC
2023-05-10 07:52:45 +03:00
parent a617d64882
commit 762265eab5
22 changed files with 40 additions and 41 deletions

View File

@@ -165,7 +165,7 @@ class VQModel(pl.LightningModule):
def validation_step(self, batch, batch_idx):
log_dict = self._validation_step(batch, batch_idx)
with self.ema_scope():
log_dict_ema = self._validation_step(batch, batch_idx, suffix="_ema")
self._validation_step(batch, batch_idx, suffix="_ema")
return log_dict
def _validation_step(self, batch, batch_idx, suffix=""):