From 62226469ec522a17b5c8db39311cb9eb72c2246c Mon Sep 17 00:00:00 2001 From: Jonas Kulhanek Date: Mon, 26 Jun 2023 16:29:32 +0200 Subject: [PATCH] Fix typing error --- nerfstudio/data/datamanagers/base_datamanager.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/nerfstudio/data/datamanagers/base_datamanager.py b/nerfstudio/data/datamanagers/base_datamanager.py index 8c052b88c..e017c73c8 100644 --- a/nerfstudio/data/datamanagers/base_datamanager.py +++ b/nerfstudio/data/datamanagers/base_datamanager.py @@ -519,6 +519,7 @@ class VanillaDataManager(DataManager, Generic[TDataset]): @torch.compile def _next_train_compiled(self, image_batch): + assert self.train_pixel_sampler is not None batch = self.train_pixel_sampler.sample(image_batch) ray_indices = batch["indices"] ray_bundle = self.train_ray_generator(ray_indices) @@ -534,6 +535,7 @@ class VanillaDataManager(DataManager, Generic[TDataset]): @torch.compile def _next_eval_compiled(self, image_batch): + assert self.eval_pixel_sampler is not None batch = self.eval_pixel_sampler.sample(image_batch) ray_indices = batch["indices"] ray_bundle = self.eval_ray_generator(ray_indices)