From cad676fc423efeafde65f03e4815248f2d357011 Mon Sep 17 00:00:00 2001
From: aktersnurra <gustaf.rydholm@gmail.com>
Date: Sun, 15 Nov 2020 16:24:12 +0100
Subject: Small changes.

---
 .../tests/support/create_emnist_lines_support_files.py               | 5 ++++-
 src/text_recognizer/tests/support/create_iam_lines_support_files.py  | 5 ++++-
 2 files changed, 8 insertions(+), 2 deletions(-)

diff --git a/src/text_recognizer/tests/support/create_emnist_lines_support_files.py b/src/text_recognizer/tests/support/create_emnist_lines_support_files.py
index b200ff5..4496e40 100644
--- a/src/text_recognizer/tests/support/create_emnist_lines_support_files.py
+++ b/src/text_recognizer/tests/support/create_emnist_lines_support_files.py
@@ -2,6 +2,8 @@
 from pathlib import Path
 import shutil
 
+import numpy as np
+
 from text_recognizer.datasets import EmnistLinesDataset
 import text_recognizer.util as util
 
@@ -10,6 +12,7 @@ SUPPORT_DIRNAME = Path(__file__).parents[0].resolve() / "emnist_lines"
 
 
 def create_emnist_lines_support_files() -> None:
+    """Create EMNIST Lines test images."""
     shutil.rmtree(SUPPORT_DIRNAME, ignore_errors=True)
     SUPPORT_DIRNAME.mkdir()
 
@@ -27,7 +30,7 @@ def create_emnist_lines_support_files() -> None:
                 for label in np.argmax(target[1:], dim=-1).flatten()
             )
             .stip()
-            .strip(self.mapper.pad_token)
+            .strip(dataset.mapper.pad_token)
         )
 
         print(label)
diff --git a/src/text_recognizer/tests/support/create_iam_lines_support_files.py b/src/text_recognizer/tests/support/create_iam_lines_support_files.py
index 15d0e4e..bb568ee 100644
--- a/src/text_recognizer/tests/support/create_iam_lines_support_files.py
+++ b/src/text_recognizer/tests/support/create_iam_lines_support_files.py
@@ -2,6 +2,8 @@
 from pathlib import Path
 import shutil
 
+import numpy as np
+
 from text_recognizer.datasets import IamLinesDataset
 import text_recognizer.util as util
 
@@ -10,6 +12,7 @@ SUPPORT_DIRNAME = Path(__file__).parents[0].resolve() / "iam_lines"
 
 
 def create_emnist_lines_support_files() -> None:
+    """Create IAM Lines test images."""
     shutil.rmtree(SUPPORT_DIRNAME, ignore_errors=True)
     SUPPORT_DIRNAME.mkdir()
 
@@ -27,7 +30,7 @@ def create_emnist_lines_support_files() -> None:
                 for label in np.argmax(target[1:], dim=-1).flatten()
             )
             .stip()
-            .strip(self.mapper.pad_token)
+            .strip(dataset.mapper.pad_token)
         )
 
         print(label)
-- 
cgit v1.2.3-70-g09d2