From 98ba548ed7d80bce20a3cdc39b511e423b5cfa23 Mon Sep 17 00:00:00 2001 From: acse-pms122 Date: Wed, 24 Jan 2024 19:00:39 +0100 Subject: [PATCH] fix pyright issues --- seathru/seathru_field.py | 4 ++-- seathru/seathru_model.py | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/seathru/seathru_field.py b/seathru/seathru_field.py index 28ac1c1..ce86364 100644 --- a/seathru/seathru_field.py +++ b/seathru/seathru_field.py @@ -186,8 +186,8 @@ def get_density(self, ray_samples: RaySamples) -> Tuple[Tensor, Tensor]: density = density * selector[..., None] return density, bottleneck_vector - def get_outputs( - self, ray_samples: RaySamples, density_embedding: Tensor + def get_outputs( # type: ignore + self, ray_samples: RaySamples, density_embedding: Optional[Tensor] = None ) -> Dict[SeathruHeadNames, Tensor]: """Compute outputs of object and medium networks (except object density). diff --git a/seathru/seathru_model.py b/seathru/seathru_model.py index c0cf34b..24cc695 100644 --- a/seathru/seathru_model.py +++ b/seathru/seathru_model.py @@ -154,7 +154,7 @@ class SeathruModel(Model): config: SeaThru-NeRF configuration to instantiate the model with. """ - config: SeathruModelConfig + config: SeathruModelConfig # type: ignore def populate_modules(self): """Setup the fields and modules.""" @@ -349,7 +349,7 @@ def bias(x, b): return callbacks - def get_outputs(self, ray_bundle: RayBundle) -> Dict[str, torch.Tensor]: + def get_outputs(self, ray_bundle: RayBundle) -> Dict[str, torch.Tensor]: # type: ignore """Get outputs from the model. Args: