TNO Intern

Commit 0c008efc authored by Florian Knappers's avatar Florian Knappers
Browse files

finished. Next up- replacing all usages of old version

parent 2f3fe597

pythermogis/pytg3/doublet.py

+2 −4
Original line number Diff line number Diff line
@@ -72,7 +72,7 @@ class ThermoGISDoublet: 
            mask_value = 0.0 if p_values is not None else np.nan



        if p_values is not None:

            thickness, transmissivity = self._expand_p_values(p_values)

            thickness, transmissivity = self._resolve_stochastic_properties(p_values)

        else:

            thickness = self.aquifer.thickness

            transmissivity = self.aquifer.transmissivity
@@ -135,10 +135,8 @@ class ThermoGISDoublet: 
            prd_temp=output_arrays[13],

            transmissivity_with_ntg=transmissivity_with_ntg,

        )

    def _expand_p_values(self, p_values: list[float]) -> tuple[xr.DataArray, xr.DataArray]:

        if not isinstance(self.aquifer, StochasticAquifer):

            raise TypeError("p_values requires a StochasticAquifer")



    def _resolve_stochastic_properties(self, p_values: list[float]) -> tuple[xr.DataArray, xr.DataArray]:

        p_values_da = xr.DataArray(

            data=p_values, dims=["p_value"], coords={"p_value": p_values}

        )