Skip to content

Commit

Permalink
fix(pyspark): ensure that the output of zip matches the expected ibis…
Browse files Browse the repository at this point in the history
… schema
  • Loading branch information
cpcloud committed Apr 25, 2024
1 parent e66cdbe commit cc1a40e
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion ibis/backends/pyspark/compiler.py
Original file line number Diff line number Diff line change
Expand Up @@ -349,7 +349,7 @@ def visit_MapGet(self, op, *, arg, key, default):
return self.if_(self.f.map_contains_key(arg, key), arg[key], default)

def visit_ArrayZip(self, op, *, arg):
return self.f.arrays_zip(*arg)
return self.cast(self.f.arrays_zip(*arg), op.dtype)

def visit_ArrayMap(self, op, *, arg, body, param):
param = sge.Identifier(this=param)
Expand Down

0 comments on commit cc1a40e

Please sign in to comment.