Skip to content

Commit ee4604a

Browse files
Casting Cs_w to float
1 parent 32d7456 commit ee4604a

File tree

1 file changed

+2
-6
lines changed

1 file changed

+2
-6
lines changed

parcels/compilation/codegenerator.py

Lines changed: 2 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -830,9 +830,7 @@ def visit_FieldEvalNode(self, node):
830830
args = self._check_FieldSamplingArguments(node.args.ccode)
831831
if "croco" in node.field.obj.gridindexingtype and node.field.obj.name != "H" and node.field.obj.name != "Zeta":
832832
# Get Cs_w values directly from fieldset (since they are 1D in vertical only)
833-
Cs_w = [
834-
self.fieldset.Cs_w.data[0][zi][0][0].astype(float) for zi in range(self.fieldset.Cs_w.data.shape[1])
835-
]
833+
Cs_w = [float(self.fieldset.Cs_w.data[0][zi][0][0]) for zi in range(self.fieldset.Cs_w.data.shape[1])]
836834
statements_croco = [
837835
c.Statement(f"float cs_w[] = {*Cs_w, }".replace("(", "{").replace(")", "}")),
838836
c.Statement(
@@ -861,9 +859,7 @@ def visit_VectorFieldEvalNode(self, node):
861859
args = self._check_FieldSamplingArguments(node.args.ccode)
862860
if "3DSigma" in node.field.obj.vector_type:
863861
# Get Cs_w values directly from fieldset (since they are 1D in vertical only)
864-
Cs_w = [
865-
self.fieldset.Cs_w.data[0][zi][0][0].astype(float) for zi in range(self.fieldset.Cs_w.data.shape[1])
866-
]
862+
Cs_w = [float(self.fieldset.Cs_w.data[0][zi][0][0]) for zi in range(self.fieldset.Cs_w.data.shape[1])]
867863
statements_croco = [
868864
c.Statement(f"float cs_w[] = {*Cs_w, }".replace("(", "{").replace(")", "}")),
869865
c.Statement(

0 commit comments

Comments
 (0)