fix pyright issues

AkerBP · Jan 24, 2024 · 98ba548 · 98ba548
1 parent aa4c0b5
commit 98ba548
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git 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
@@ -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: