summaryrefslogtreecommitdiff
path: root/become_yukarin/updater
diff options
context:
space:
mode:
Diffstat (limited to 'become_yukarin/updater')
-rw-r--r--become_yukarin/updater/sr_updater.py2
-rw-r--r--become_yukarin/updater/updater.py2
2 files changed, 2 insertions, 2 deletions
diff --git a/become_yukarin/updater/sr_updater.py b/become_yukarin/updater/sr_updater.py
index 6e2b400..88f4bb3 100644
--- a/become_yukarin/updater/sr_updater.py
+++ b/become_yukarin/updater/sr_updater.py
@@ -14,7 +14,7 @@ class SRUpdater(chainer.training.StandardUpdater):
discriminator: SRDiscriminator,
*args,
**kwargs,
- ):
+ ) -> None:
super().__init__(*args, **kwargs)
self.loss_config = loss_config
self.predictor = predictor
diff --git a/become_yukarin/updater/updater.py b/become_yukarin/updater/updater.py
index ef77e77..8dcb215 100644
--- a/become_yukarin/updater/updater.py
+++ b/become_yukarin/updater/updater.py
@@ -17,7 +17,7 @@ class Updater(chainer.training.StandardUpdater):
discriminator: Discriminator = None,
*args,
**kwargs,
- ):
+ ) -> None:
super().__init__(*args, **kwargs)
self.loss_config = loss_config
self.predictor = predictor