black and flake my friends

onmt/transforms/transform.py

@@ -264,13 +264,16 @@ def _repr_args(self):
 def make_transforms(opts, transforms_cls, vocabs):
     """Build transforms in `transforms_cls` with vocab of `fields`."""
     transforms = {}
-    for name, transform_cls in transforms_cls.items():
-        if transform_cls.require_vocab() and vocabs is None:
-            logger.warning(f"{transform_cls.__name__} require vocab to apply, skip it.")
-            continue
-        transform_obj = transform_cls(opts)
-        transform_obj.warm_up(vocabs)
-        transforms[name] = transform_obj
+    if transforms_cls:
+        for name, transform_cls in transforms_cls.items():
+            if transform_cls.require_vocab() and vocabs is None:
+                logger.warning(
+                    f"{transform_cls.__name__} require vocab to apply, skip it."
+                )
+                continue
+            transform_obj = transform_cls(opts)
+            transform_obj.warm_up(vocabs)
+            transforms[name] = transform_obj
     return transforms
 
 

onmt/utils/scoring_utils.py

@@ -5,7 +5,7 @@
 from onmt.opts import translate_opts
 from onmt.constants import CorpusTask
 from onmt.inputters.dynamic_iterator import build_dynamic_dataset_iter
-from onmt.transforms import get_transforms_cls, make_transforms, TransformPipe
+from onmt.transforms import get_transforms_cls
 
 
 class ScoringPreparator:
@@ -19,6 +19,8 @@ def __init__(self, vocabs, opt):
         if self.opt.dump_preds is not None:
             if not os.path.exists(self.opt.dump_preds):
                 os.makedirs(self.opt.dump_preds)
+        self.transforms = None
+        self.transforms_cls = None
 
     def warm_up(self, transforms):
         self.transforms = transforms
@@ -78,7 +80,7 @@ def translate(self, model, gpu_rank, step):
 
         # Reinstantiate the validation iterator
 
-        #transforms_cls = get_transforms_cls(model_opt._all_transform)
+        # transforms_cls = get_transforms_cls(model_opt._all_transform)
         model_opt.num_workers = 0
         model_opt.tgt = None