From affb689164cd2dce6c57d23d8ce9852e620f567e Mon Sep 17 00:00:00 2001 From: Alexander Raistrick Date: Fri, 1 Nov 2024 15:29:39 -0400 Subject: [PATCH] Change generate_nature --debug behavior to match generate_indoors --- infinigen_examples/generate_indoors.py | 1 + infinigen_examples/generate_nature.py | 20 +++++++++++--------- 2 files changed, 12 insertions(+), 9 deletions(-) diff --git a/infinigen_examples/generate_indoors.py b/infinigen_examples/generate_indoors.py index dbf20108..cbf1f14b 100644 --- a/infinigen_examples/generate_indoors.py +++ b/infinigen_examples/generate_indoors.py @@ -552,6 +552,7 @@ def main(args): parser.add_argument("-d", "--debug", type=str, nargs="*", default=None) args = init.parse_args_blender(parser) + logging.getLogger("infinigen").setLevel(logging.INFO) logging.getLogger("infinigen.core.nodes.node_wrangler").setLevel(logging.CRITICAL) diff --git a/infinigen_examples/generate_nature.py b/infinigen_examples/generate_nature.py index d04a1b79..a75dae7d 100644 --- a/infinigen_examples/generate_nature.py +++ b/infinigen_examples/generate_nature.py @@ -1023,16 +1023,18 @@ def main(args): "e.g. --gin_param module_1.a=2 module_2.b=3", ) parser.add_argument("--task_uniqname", type=str, default=None) - parser.add_argument( - "-d", - "--debug", - action="store_const", - dest="loglevel", - const=logging.DEBUG, - default=logging.INFO, - ) + parser.add_argument("-d", "--debug", type=str, nargs="*", default=None) args = init.parse_args_blender(parser) - logging.getLogger("infinigen").setLevel(args.loglevel) + + logging.getLogger("infinigen").setLevel(logging.INFO) + logging.getLogger("infinigen.core.nodes.node_wrangler").setLevel(logging.CRITICAL) + + if args.debug is not None: + for name in logging.root.manager.loggerDict: + if not name.startswith("infinigen"): + continue + if len(args.debug) == 0 or any(name.endswith(x) for x in args.debug): + logging.getLogger(name).setLevel(logging.DEBUG) main(args)