Fix up transport 2

illinois-ceesd · Sep 7, 2024 · 7800e67 · 7800e67
1 parent 309c424
commit 7800e67
Showing 1 changed file with 8 additions and 8 deletions.
diff --git a/mirgecom/transport.py b/mirgecom/transport.py
@@ -167,13 +167,13 @@ def bulk_viscosity(self, cv: ConservedVars,
                        dv: Optional[GasDependentVars] = None,
                        eos: Optional[GasEOS] = None) -> DOFArray:
         r"""Get the bulk viscosity for the gas, $\mu_{B}$."""
-        return self._mu_bulk*(0*cv.mass + 1.0)
+        return self._mu_bulk*(0*cv.mass + 1.0)  # type: ignore
 
     def viscosity(self, cv: ConservedVars,
                   dv: Optional[GasDependentVars] = None,
                   eos: Optional[GasEOS] = None) -> DOFArray:
         r"""Get the gas dynamic viscosity, $\mu$."""
-        return self._mu*(0*cv.mass + 1.0)
+        return self._mu*(0*cv.mass + 1.0)  # type: ignore
 
     def volume_viscosity(self, cv: ConservedVars,
                          dv: Optional[GasDependentVars] = None,
@@ -187,19 +187,19 @@ def volume_viscosity(self, cv: ConservedVars,
             \lambda = \left(\mu_{B} - \frac{2\mu}{3}\right)
 
         """
-        return (self._mu_bulk - 2 * self._mu / 3)*(0*cv.mass + 1.0)
+        return (self._mu_bulk - 2 * self._mu / 3)*(0*cv.mass + 1.0)  # type: ignore
 
     def thermal_conductivity(self, cv: ConservedVars,
                              dv: Optional[GasDependentVars] = None,
                              eos: Optional[GasEOS] = None) -> DOFArray:
         r"""Get the gas thermal_conductivity, $\kappa$."""
-        return self._kappa*(0*cv.mass + 1.0)
+        return self._kappa*(0*cv.mass + 1.0)  # type: ignore
 
     def species_diffusivity(self, cv: ConservedVars,
                             dv: Optional[GasDependentVars] = None,
                             eos: Optional[GasEOS] = None) -> np.ndarray:
         r"""Get the vector of species diffusivities, ${d}_{\alpha}$."""
-        return self._d_alpha*(0*cv.mass + 1.0)
+        return self._d_alpha*(0*cv.mass + 1.0)  # type: ignore
 
 
 class PowerLawTransport(TransportModel):
@@ -318,7 +318,7 @@ def species_diffusivity(self, cv: ConservedVars,  # type: ignore[override]
             return (self._sigma * self.viscosity(cv, dv)/(
                 cv.mass*self._lewis*eos.gamma(cv, dv.temperature))
             )
-        return self._d_alpha*(0*cv.mass + 1.)
+        return self._d_alpha*(0*cv.mass + 1.)  # type: ignore
 
 
 class MixtureAveragedTransport(TransportModel):
@@ -576,7 +576,7 @@ def thermal_conductivity(self, cv: ConservedVars,  # type: ignore[override]
 
     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._physical_transport.species_diffusivity(cv, dv, eos)
 
@@ -696,7 +696,7 @@ def thermal_conductivity(self, cv: ConservedVars,  # type: ignore[override]
 
     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._physical_transport.species_diffusivity(cv, dv, eos)