Import max_utils directly inside main()

amepas · amepas · commit f68554780290 · 2026-07-16T15:38:34.000Z
diff --git a/src/maxdiffusion/generate_flux2klein.py b/src/maxdiffusion/generate_flux2klein.py
@@ -561,6 +561,7 @@ def handle_exception(exc_type, exc_value, exc_traceback):
   ]
   default_args.extend(custom_overrides)
 
+  from maxdiffusion import max_utils
   pyconfig.initialize(default_args)
   max_utils.ensure_machinelearning_job_runs(pyconfig.config)
   run(pyconfig.config)

Original file line number	Diff line number	Diff line change
`@@ -561,6 +561,7 @@ def handle_exception(exc_type, exc_value, exc_traceback):`
`561`	`561`	`]`
`562`	`562`	`default_args.extend(custom_overrides)`
`563`	`563`
	`564`	`+ from maxdiffusion import max_utils`
`564`	`565`	`pyconfig.initialize(default_args)`
`565`	`566`	`max_utils.ensure_machinelearning_job_runs(pyconfig.config)`
`566`	`567`	`run(pyconfig.config)`