def get_new_model_name(model_checkpoint: str, model_name: str = None) -> str: if model_name is None: old_version_number = int(model_checkpoint[-2:]) new_version_number = str(old_version_number + 1).zfill(2) model_name = f"{model_checkpoint[:-2]}{new_version_number}" elif not model_name[-2:].isnumeric(): model_name = model_name + "_00" return model_name