Fix the type of the value returned by the main method.

Author: foxik

labs/11/vae.py

@@ -127,7 +127,7 @@ def generate(self, epoch: int, logs: dict[str, float]) -> None:
             self.get_tb_writer("train").add_image("images", grid, epoch)
 
 
-def main(args: argparse.Namespace) -> dict[str, float]:
+def main(args: argparse.Namespace) -> float:
     # Set the random seed and the number of threads.
     npfl138.startup(args.seed, args.threads)
     npfl138.global_keras_initializers()