From 88b253f506af1775a80ceb0607c8af8164fda70f Mon Sep 17 00:00:00 2001 From: Matias Turkulainen <30566358+maturk@users.noreply.github.com> Date: Sat, 10 Jun 2023 04:21:17 +0300 Subject: [PATCH] Add kwargs to dataparser outputs (#2053) add kwargs to dataparser outputs --- nerfstudio/data/dataparsers/base_dataparser.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/nerfstudio/data/dataparsers/base_dataparser.py b/nerfstudio/data/dataparsers/base_dataparser.py index 1260833b9..80fab739f 100644 --- a/nerfstudio/data/dataparsers/base_dataparser.py +++ b/nerfstudio/data/dataparsers/base_dataparser.py @@ -141,26 +141,28 @@ class DataParser: self.config = config @abstractmethod - def _generate_dataparser_outputs(self, split: str = "train") -> DataparserOutputs: + def _generate_dataparser_outputs(self, split: str = "train", **kwargs: Optional[Dict]) -> DataparserOutputs: """Abstract method that returns the dataparser outputs for the given split. Args: split: Which dataset split to generate (train/test). + kwargs: kwargs for generating dataparser outputs. Returns: DataparserOutputs containing data for the specified dataset and split """ - def get_dataparser_outputs(self, split: str = "train") -> DataparserOutputs: + def get_dataparser_outputs(self, split: str = "train", **kwargs: Optional[Dict]) -> DataparserOutputs: """Returns the dataparser outputs for the given split. Args: split: Which dataset split to generate (train/test). + kwargs: kwargs for generating dataparser outputs. Returns: DataparserOutputs containing data for the specified dataset and split """ - dataparser_outputs = self._generate_dataparser_outputs(split) + dataparser_outputs = self._generate_dataparser_outputs(split, **kwargs) return dataparser_outputs