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