From acd126c1ef2086b2a2904c0102e552eed07ca2ed Mon Sep 17 00:00:00 2001 From: Brent Yi Date: Mon, 19 Sep 2022 00:28:06 -0700 Subject: [PATCH] Set dcargs accent color (#465) * Set dcargs accent color * Bump required dcargs version * color -> colour * Bump dcargs to 0.3.6 --- nerfactory/viewer/app/run_deploy.py | 1 + pyproject.toml | 2 +- scripts/completions/configure.py | 1 + scripts/debugging/run_actions.py | 1 + scripts/docs/add_nb_tags.py | 1 + scripts/docs/build_docs.py | 1 + scripts/downloads/download_data.py | 6 ++++++ scripts/process_data.py | 1 + scripts/run_eval.py | 1 + scripts/run_train.py | 1 + scripts/viewer/view_dataset.py | 1 + 11 files changed, 16 insertions(+), 1 deletion(-) diff --git a/nerfactory/viewer/app/run_deploy.py b/nerfactory/viewer/app/run_deploy.py index 64893cf19..53f9306d0 100644 --- a/nerfactory/viewer/app/run_deploy.py +++ b/nerfactory/viewer/app/run_deploy.py @@ -135,4 +135,5 @@ def main( if __name__ == "__main__": + dcargs.extras.set_accent_color("bright_yellow") dcargs.cli(main) diff --git a/pyproject.toml b/pyproject.toml index f85bc3665..1422e2ec8 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -18,7 +18,7 @@ dependencies = [ "aiortc==1.3.2", "av==9.2.0", "black[jupyter]==22.3.0", - "dcargs>=0.3.1", + "dcargs>=0.3.6", "gdown==4.5.1", "ninja==1.10.2.3", "functorch==0.2.1", diff --git a/scripts/completions/configure.py b/scripts/completions/configure.py index 8860bba96..23f57aa55 100755 --- a/scripts/completions/configure.py +++ b/scripts/completions/configure.py @@ -246,4 +246,5 @@ def main( if __name__ == "__main__": + dcargs.extras.set_accent_color("bright_yellow") dcargs.cli(main, description=__doc__) diff --git a/scripts/debugging/run_actions.py b/scripts/debugging/run_actions.py index a195eae3b..9e28e9762 100755 --- a/scripts/debugging/run_actions.py +++ b/scripts/debugging/run_actions.py @@ -87,4 +87,5 @@ def run_code_checks(continue_on_fail: bool = False): if __name__ == "__main__": + dcargs.extras.set_accent_color("bright_yellow") dcargs.cli(run_code_checks) diff --git a/scripts/docs/add_nb_tags.py b/scripts/docs/add_nb_tags.py index a41ddf783..ea4468c8f 100755 --- a/scripts/docs/add_nb_tags.py +++ b/scripts/docs/add_nb_tags.py @@ -76,4 +76,5 @@ def main(check: bool = False): if __name__ == "__main__": + dcargs.extras.set_accent_color("bright_yellow") dcargs.cli(main) diff --git a/scripts/docs/build_docs.py b/scripts/docs/build_docs.py index ff217d48e..21606dc16 100755 --- a/scripts/docs/build_docs.py +++ b/scripts/docs/build_docs.py @@ -48,4 +48,5 @@ def main(clean_cache: bool = False): if __name__ == "__main__": + dcargs.extras.set_accent_color("bright_yellow") dcargs.cli(main) diff --git a/scripts/downloads/download_data.py b/scripts/downloads/download_data.py index 07dc4265f..61e643f2b 100755 --- a/scripts/downloads/download_data.py +++ b/scripts/downloads/download_data.py @@ -75,4 +75,10 @@ def main(dataset: Literal["blender", "friends"]): if __name__ == "__main__": + try: + # Requires dcargs>=0.3.3. + dcargs.extras.set_accent_color("bright_yellow") + except AttributeError: + pass + dcargs.cli(main) diff --git a/scripts/process_data.py b/scripts/process_data.py index 482f27c98..f9d6ebffc 100755 --- a/scripts/process_data.py +++ b/scripts/process_data.py @@ -441,4 +441,5 @@ def main( if __name__ == "__main__": + dcargs.extras.set_accent_color("bright_yellow") dcargs.cli(main) diff --git a/scripts/run_eval.py b/scripts/run_eval.py index 2cc34fc55..effc1509f 100755 --- a/scripts/run_eval.py +++ b/scripts/run_eval.py @@ -255,4 +255,5 @@ if __name__ == "__main__": # operations implemented by each seem fairly universal; ideally the checkpoint # loading logic, for example, would be the same as what's used for loading a # checkpoint when resuming a training run. + dcargs.extras.set_accent_color("bright_yellow") dcargs.cli(Union[ComputePSNR, RenderTrajectory]).main() diff --git a/scripts/run_train.py b/scripts/run_train.py index 2186a89eb..783e979ed 100755 --- a/scripts/run_train.py +++ b/scripts/run_train.py @@ -212,4 +212,5 @@ def main(config: cfg.Config) -> None: if __name__ == "__main__": # Choose a base configuration and override values. + dcargs.extras.set_accent_color("bright_yellow") main(dcargs.cli(AnnotatedBaseConfigUnion)) diff --git a/scripts/viewer/view_dataset.py b/scripts/viewer/view_dataset.py index f3cf6aa28..3f894a2d9 100755 --- a/scripts/viewer/view_dataset.py +++ b/scripts/viewer/view_dataset.py @@ -39,6 +39,7 @@ def main(config: cfg.Config) -> None: if __name__ == "__main__": console = Console(width=120) + dcargs.extras.set_accent_color("bright_yellow") instantiated_config = dcargs.cli(AnnotatedBaseConfigUnion) if instantiated_config.trainer.load_config: logging.info(f"Loading pre-set config from: {instantiated_config.trainer.load_config}")