mirror of
https://github.com/nerfstudio-project/nerfstudio.git
synced 2026-11-10 18:17:31 +08:00
8e4ae6bebf
* Allow black/white background color in Nerfacto * Use white background in Nerfacto Blender Benchmark * Fixed unused import * Updated index for license headers script
97 lines
3.4 KiB
Python
97 lines
3.4 KiB
Python
#!/usr/bin/env python
|
|
"""Simple yaml debugger"""
|
|
import subprocess
|
|
import sys
|
|
|
|
import tyro
|
|
import yaml
|
|
from rich.console import Console
|
|
from rich.style import Style
|
|
|
|
CONSOLE = Console(width=120)
|
|
|
|
LOCAL_TESTS = ["Run license checks", "Run isort", "Run Black", "Python Pylint", "Test with pytest"]
|
|
|
|
|
|
def run_command(command: str, continue_on_fail: bool = False) -> bool:
|
|
"""Run a command kill actions if it fails
|
|
|
|
Args:
|
|
command: command to run
|
|
continue_on_fail: whether to continue running commands if the current one fails.
|
|
"""
|
|
ret_code = subprocess.call(command, shell=True)
|
|
if ret_code != 0:
|
|
CONSOLE.print(f"[bold red]Error: `{command}` failed.")
|
|
if not continue_on_fail:
|
|
sys.exit(1)
|
|
return ret_code == 0
|
|
|
|
|
|
def run_github_actions_file(filename: str, continue_on_fail: bool = False):
|
|
"""Run a github actions file locally.
|
|
|
|
Args:
|
|
filename: Which yml github actions file to run.
|
|
continue_on_fail: Whether or not to continue running actions commands if the current one fails
|
|
"""
|
|
with open(filename, "rb") as f:
|
|
my_dict = yaml.safe_load(f)
|
|
steps = my_dict["jobs"]["build"]["steps"]
|
|
|
|
success = True
|
|
|
|
for step in steps:
|
|
if "name" in step and step["name"] in LOCAL_TESTS:
|
|
compressed = step["run"].replace("\n", ";").replace("\\", "")
|
|
compressed = compressed.replace("--check", "")
|
|
curr_command = f"{compressed}"
|
|
|
|
CONSOLE.line()
|
|
CONSOLE.rule(f"[bold green]Running: {curr_command}")
|
|
success = success and run_command(curr_command, continue_on_fail=continue_on_fail)
|
|
else:
|
|
skip_name = step["name"] if "name" in step else step["uses"]
|
|
CONSOLE.print(f"Skipping {skip_name}")
|
|
|
|
# Add checks for building documentation
|
|
CONSOLE.line()
|
|
CONSOLE.rule("[bold green]Adding notebook documentation metadata")
|
|
success = success and run_command("python scripts/docs/add_nb_tags.py", continue_on_fail=continue_on_fail)
|
|
CONSOLE.line()
|
|
CONSOLE.rule("[bold green]Building Documentation")
|
|
success = success and run_command("cd docs/; make html SPHINXOPTS='-W;'", continue_on_fail=continue_on_fail)
|
|
|
|
if success:
|
|
CONSOLE.line()
|
|
CONSOLE.rule(characters="=")
|
|
CONSOLE.print("[bold green]:TADA: :TADA: :TADA: ALL CHECKS PASSED :TADA: :TADA: :TADA:", justify="center")
|
|
CONSOLE.rule(characters="=")
|
|
else:
|
|
CONSOLE.line()
|
|
CONSOLE.rule(characters="=", style=Style(color="red"))
|
|
CONSOLE.print("[bold red]:skull: :skull: :skull: ERRORS FOUND :skull: :skull: :skull:", justify="center")
|
|
CONSOLE.rule(characters="=", style=Style(color="red"))
|
|
|
|
|
|
def run_code_checks(continue_on_fail: bool = False):
|
|
"""Run a github actions file locally.
|
|
|
|
Args:
|
|
continue_on_fail: Whether or not to continue running actions commands if the current one fails
|
|
"""
|
|
# core code checks
|
|
run_github_actions_file(filename=".github/workflows/core_code_checks.yml", continue_on_fail=continue_on_fail)
|
|
# viewer build and deployment
|
|
# run_github_actions_file(filename=".github/workflows/viewer_build_deploy.yml", continue_on_fail=continue_on_fail)
|
|
|
|
|
|
def entrypoint():
|
|
"""Entrypoint for use with pyproject scripts."""
|
|
tyro.extras.set_accent_color("bright_yellow")
|
|
tyro.cli(run_code_checks)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
entrypoint()
|