Loose assert condition for datamanger in get_average_eval_image_metrics() (#3310)

* loosen datamanger assert

* linter
This commit is contained in:
Congrong Xu authored and GitHub committed 2024-07-16 15:16:38 -04:00
1 parent 5945851598
commit 7736539cc7
1 file changed
+4 -6
+4 -6
View File
@@ -35,9 +35,7 @@ from torch.nn import Parameter
from torch.nn.parallel import DistributedDataParallel as DDP
from nerfstudio.configs.base_config import InstantiateConfig
from nerfstudio.data.datamanagers.base_datamanager import DataManager, DataManagerConfig, VanillaDataManager
from nerfstudio.data.datamanagers.full_images_datamanager import FullImageDatamanager
from nerfstudio.data.datamanagers.parallel_datamanager import ParallelDataManager
from nerfstudio.data.datamanagers.base_datamanager import DataManager, DataManagerConfig
from nerfstudio.engine.callbacks import TrainingCallback, TrainingCallbackAttributes
from nerfstudio.models.base_model import Model, ModelConfig
from nerfstudio.utils import profiler
@@ -424,9 +422,9 @@ class VanillaPipeline(Pipeline):
self, step: Optional[int] = None, output_path: Optional[Path] = None, get_std: bool = False
):
"""Get the average metrics for evaluation images."""
"""Get the average metrics for evaluation images."""
assert isinstance(self.datamanager, (VanillaDataManager, ParallelDataManager, FullImageDatamanager))
assert hasattr(
self.datamanager, "fixed_indices_eval_dataloader"
), "datamanager must have 'fixed_indices_eval_dataloader' attribute"
image_prefix = "eval"
return self.get_average_image_metrics(
self.datamanager.fixed_indices_eval_dataloader, image_prefix, step, output_path, get_std