Avoids running test_train.py on external methods (#1882)

removes testing on external methods
This commit is contained in:
Ethan Weber authored and GitHub committed 2023-05-06 12:19:26 -07:00
1 parent d847949fe9
commit 7537551f38
1 file changed
+4 -3
+4 -3
View File
@@ -83,6 +83,7 @@ descriptions = {
"nerfplayer-nerfacto": "NeRFPlayer with nerfacto backbone.",
"nerfplayer-ngp": "NeRFPlayer with InstantNGP backbone.",
"neus": "Implementation of NeuS. (slow)",
"neus-facto": "Implementation of NeuS-Facto. (slow)",
}
method_configs["nerfacto"] = TrainerConfig(
@@ -589,12 +590,12 @@ method_configs["neus-facto"] = TrainerConfig(
)
external_methods, external_descriptions = discover_methods()
method_configs.update(external_methods)
descriptions.update(external_descriptions)
all_methods = {**method_configs, **external_methods}
all_descriptions = {**descriptions, **external_descriptions}
AnnotatedBaseConfigUnion = tyro.conf.SuppressFixed[ # Don't show unparseable (fixed) arguments in helptext.
tyro.conf.FlagConversionOff[
tyro.extras.subcommand_type_from_defaults(defaults=method_configs, descriptions=descriptions)
tyro.extras.subcommand_type_from_defaults(defaults=all_methods, descriptions=all_descriptions)
]
]
"""Union[] type over config types, annotated with default instances for use with