home · contact · privacy
Postpone overhead-y import until after validating/parsing parameters.
authorChristian Heller <c.heller@plomlompom.de>
Mon, 19 Aug 2024 22:08:40 +0000 (00:08 +0200)
committerChristian Heller <c.heller@plomlompom.de>
Mon, 19 Aug 2024 22:08:40 +0000 (00:08 +0200)
stable.py

index cdd6667a3df57815b67c6bdb4f74a1d4139f310e..634add345326e22539b8e8867646bd9e1b68551c 100755 (executable)
--- a/stable.py
+++ b/stable.py
@@ -3,7 +3,6 @@ from sys import argv, exit as sys_exit
 from os.path import dirname, basename, splitext, join as path_join, exists
 from argparse import ArgumentParser, ArgumentDefaultsHelpFormatter
 from random import randint
-from stable.core import ImageMaker
 
 
 DEFAULT_SCHEDULER = 'EulerAncestralDiscreteScheduler'
@@ -15,7 +14,8 @@ def save_path(count: int) -> str:
 
 
 def parse_args():
-    parser = ArgumentParser(add_help=False, formatter_class=ArgumentDefaultsHelpFormatter)
+    parser = ArgumentParser(add_help=False,
+                            formatter_class=ArgumentDefaultsHelpFormatter)
     parser.add_argument('-m', '--model',
                         help='model filename (-P will pre prefixed, but may '
                         'also be full path on its own)')
@@ -88,6 +88,8 @@ def parse_args():
 
 
 args = parse_args()
+# pylint: disable=wrong-import-position
+from stable.core import ImageMaker  # noqa: E402
 model_path = f'{args.model_path_prefix}{args.model}'
 if args.list_schedulers:
     maker = ImageMaker(model_path)