rewrite

engine/base/i2i.py

@@ -64,7 +64,7 @@ class EngineKernel(object):
         self.engine = engine
 
     def build_models(self) -> (dict, dict):
-        raise NotImplemented
+        raise NotImplementedError
 
     def to_save(self):
         to_save = {}
@@ -73,19 +73,19 @@ class EngineKernel(object):
         return to_save
 
     def setup_after_g(self):
-        raise NotImplemented
+        raise NotImplementedError
 
     def setup_before_g(self):
-        raise NotImplemented
+        raise NotImplementedError
 
     def forward(self, batch, inference=False) -> dict:
-        raise NotImplemented
+        raise NotImplementedError
 
     def criterion_generators(self, batch, generated) -> dict:
-        raise NotImplemented
+        raise NotImplementedError
 
     def criterion_discriminators(self, batch, generated) -> dict:
-        raise NotImplemented
+        raise NotImplementedError
 
     def intermediate_images(self, batch, generated) -> dict:
         """
@@ -94,7 +94,7 @@ class EngineKernel(object):
         :param generated: dict of images
         :return: dict like: {"a": [img1, img2, ...], "b": [img3, img4, ...]}
         """
-        raise NotImplemented
+        raise NotImplementedError
 
     def change_engine(self, config, engine: Engine):
         pass