the last bug

pschwllr · Sep 13, 2018 · 8f05d7d · 8f05d7d
1 parent c009c3b
commit 8f05d7d
Showing 1 changed file with 4 additions and 5 deletions.
diff --git a/onmt/inputters/image_dataset.py b/onmt/inputters/image_dataset.py
@@ -98,11 +98,10 @@ def make_image_examples_nfeats_tpl(img_iter, img_path, img_dir,
         """
         if img_iter is None:
             if img_path is not None:
-                img_iter = \
-                    ImageDataset \
-                        .make_img_iterator_from_file(img_path,
-                                                     img_dir,
-                                                     image_channel_size)
+                img_iter = ImageDataset. \
+                    make_img_iterator_from_file(img_path,
+                                                img_dir,
+                                                image_channel_size)
             else:
                 raise ValueError("""One of 'img_iter' and 'img_path'
                                     must be not None""")