From aeac83d643aaa91c2ab310414c959447d2be5cda Mon Sep 17 00:00:00 2001 From: Aaron Wagener Date: Thu, 22 May 2025 08:29:24 -0500 Subject: [PATCH] Generate: Don't Force Player Name for Weights Files (#4943) --- Generate.py | 12 ++++++++---- 1 file changed, 8 insertions(+), 4 deletions(-) diff --git a/Generate.py b/Generate.py index 9bc8d106..f9607e32 100644 --- a/Generate.py +++ b/Generate.py @@ -224,10 +224,14 @@ def main(args=None) -> tuple[argparse.Namespace, int]: except Exception as e: raise Exception(f"Error setting {k} to {v} for player {player}") from e - if path == args.weights_file_path: # if name came from the weights file, just use base player name - erargs.name[player] = f"Player{player}" - elif player not in erargs.name: # if name was not specified, generate it from filename - erargs.name[player] = os.path.splitext(os.path.split(path)[-1])[0] + # name was not specified + if player not in erargs.name: + if path == args.weights_file_path: + # weights file, so we need to make the name unique + erargs.name[player] = f"Player{player}" + else: + # use the filename + erargs.name[player] = os.path.splitext(os.path.split(path)[-1])[0] erargs.name[player] = handle_name(erargs.name[player], player, name_counter) player += 1