def METHOD_NAME(self, save_directory, **kwargs):
    if os.path.isfile(save_directory):
        raise ValueError(f"Provided path ({save_directory}) should be a directory, not a file")
    os.makedirs(save_directory, exist_ok=True)
    return super().METHOD_NAME(save_directory, **kwargs)