Fix transport 1

illinois-ceesd · Sep 7, 2024 · 309c424 · 309c424
1 parent 6a2da92
commit 309c424
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 8 deletions.
diff --git a/mirgecom/eos.py b/mirgecom/eos.py
@@ -867,9 +867,9 @@ def get_species_source_terms(self, cv: ConservedVars, temperature: DOFArray):
         w = self.get_species_molecular_weights()
         dim = cv.dim
         species_sources = w * omega
-        rho_source = 0. * cv.mass
-        mom_source = 0. * cv.momentum
-        energy_source = 0. * cv.energy
+        rho_source = 0 * cv.mass  # type: ignore
+        mom_source = 0 * cv.momentum
+        energy_source = 0 * cv.energy  # type: ignore
 
         return make_conserved(dim, rho_source, energy_source, mom_source,
                               species_sources)

diff --git a/mirgecom/transport.py b/mirgecom/transport.py
@@ -77,9 +77,9 @@ class GasTransportVars:
     .. attribute:: species_diffusivity
     """
 
-    bulk_viscosity: np.ndarray
-    viscosity: np.ndarray
-    thermal_conductivity: np.ndarray
+    bulk_viscosity: DOFArray
+    viscosity: DOFArray
+    thermal_conductivity: DOFArray
     species_diffusivity: np.ndarray
 
 
@@ -124,7 +124,7 @@ def thermal_conductivity(self, cv: ConservedVars,
 
     def species_diffusivity(self, cv: ConservedVars,
                             dv: Optional[GasDependentVars] = None,
-                            eos: Optional[GasEOS] = None) -> DOFArray:
+                            eos: Optional[GasEOS] = None) -> np.ndarray:
         r"""Get the vector of species diffusivities, ${d}_{\alpha}$."""
         raise NotImplementedError()
 
@@ -197,7 +197,7 @@ def thermal_conductivity(self, cv: ConservedVars,
 
     def species_diffusivity(self, cv: ConservedVars,
                             dv: Optional[GasDependentVars] = None,
-                            eos: Optional[GasEOS] = None) -> DOFArray:
+                            eos: Optional[GasEOS] = None) -> np.ndarray:
         r"""Get the vector of species diffusivities, ${d}_{\alpha}$."""
         return self._d_alpha*(0*cv.mass + 1.0)