Implement ForwardRef generic interface

This commit is contained in:
Jonas Kulhanek committed 2023-03-01 11:36:50 +01:00
1 parent 8d24a3bcab
commit 77b7bfb1b1
33 files changed
+120 -89

No files matched your search

@@ -8,10 +8,9 @@ A skeleton of the local writer config is defined below.
"""nerfstudio/configs/base_config.py""""
@dataclass
class LocalWriterConfig(InstantiateConfig):
class LocalWriterConfig(InstantiateConfig[writer.LocalWriter]):
"""Local Writer config"""
_target: Type = writer.LocalWriter
enable: bool = False
stats_to_track: Tuple[writer.EventName, ...] = (
writer.EventName.ITER_TRAIN_TIME,