From e5eca28438cd17d436359f2c6eee0bb9e55d2a8b Mon Sep 17 00:00:00 2001
From: Gustaf Rydholm <gustaf.rydholm@gmail.com>
Date: Mon, 2 Aug 2021 01:01:51 +0200
Subject: Fixed wandb callback bug, add word_pieces arg to iam paragraph

---
 training/callbacks/wandb_callbacks.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

(limited to 'training/callbacks')

diff --git a/training/callbacks/wandb_callbacks.py b/training/callbacks/wandb_callbacks.py
index 451b0d5..6379cc0 100644
--- a/training/callbacks/wandb_callbacks.py
+++ b/training/callbacks/wandb_callbacks.py
@@ -94,7 +94,7 @@ class LogTextPredictions(Callback):
         super().__init__()
 
     def _log_predictions(
-        stage: str, trainer: Trainer, pl_module: LightningModule
+        self, stage: str, trainer: Trainer, pl_module: LightningModule
     ) -> None:
         """Logs the predicted text contained in the images."""
         if not self.ready:
-- 
cgit v1.2.3-70-g09d2