@@ -39,14 +39,20 @@ def update_demo(
3939 demos_cache_folder : Annotated [Path , FolderOption ("--demos-cache-folder" , "-c" )],
4040 add_rust_extension : Annotated [bool , typer .Option ("--add-rust-extension" , "-r" )] = False ,
4141 min_python_version : Annotated [str , typer .Option ("--min-python-version" )] = "3.10" ,
42- max_python_version : Annotated [str , typer .Option ("--max-python-version" )] = "3.14"
42+ max_python_version : Annotated [str , typer .Option ("--max-python-version" )] = "3.14" ,
43+ branch_override : Annotated [Optional [str ], typer .Option ("--branch-override" )] = None
4344) -> None :
4445 """Runs precommit in a generated project and matches the template to the results."""
4546 demo_name : str = get_demo_name (add_rust_extension = add_rust_extension )
4647 demo_path : Path = demos_cache_folder / demo_name
4748
48- current_branch : str = get_current_branch ()
49+ if branch_override is not None :
50+ typer .secho (f"Overriding current branch name for demo reference. Using '{ branch_override } ' instead." )
51+ current_branch : str = branch_override
52+ else :
53+ current_branch : str = get_current_branch ()
4954 template_commit : str = get_current_commit ()
55+
5056 _validate_template_main_not_checked_out (branch = current_branch )
5157 require_clean_and_up_to_date_demo_repo (demo_path = demo_path )
5258 _checkout_demo_develop_or_existing_branch (demo_path = demo_path , branch = current_branch )
0 commit comments