Skip to content

Commit 2f90812

Browse files
fixing most ns-dev-test cases
1 parent 603a5db commit 2f90812

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

nerfstudio/data/datasets/depth_dataset.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -43,7 +43,9 @@ class DepthDataset(InputDataset):
4343

4444
exclude_batch_keys_from_device = InputDataset.exclude_batch_keys_from_device + ["depth_image"]
4545

46-
def __init__(self, dataparser_outputs: DataparserOutputs, scale_factor: float = 1.0):
46+
def __init__(
47+
self, dataparser_outputs: DataparserOutputs, scale_factor: float = 1.0, cache_compressed_images: bool = False
48+
):
4749
super().__init__(dataparser_outputs, scale_factor)
4850
# if there are no depth images than we want to generate them all with zoe depth
4951

0 commit comments

Comments
 (0)