File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed
Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change 2525def df_selection (col_name , col_type ):
2626 if col_type == pa .float64 () or isinstance (col_type , pa .Decimal128Type ):
2727 return F .round (col (col_name ), lit (2 )).alias (col_name )
28- elif col_type == pa .string ():
28+ elif col_type == pa .string () or col_type == pa . string_view () :
2929 return F .trim (col (col_name )).alias (col_name )
3030 else :
3131 return col (col_name )
@@ -43,7 +43,7 @@ def load_schema(col_name, col_type):
4343def expected_selection (col_name , col_type ):
4444 if col_type == pa .int64 () or col_type == pa .int32 ():
4545 return F .trim (col (col_name )).cast (col_type ).alias (col_name )
46- elif col_type == pa .string ():
46+ elif col_type == pa .string () or col_type == pa . string_view () :
4747 return F .trim (col (col_name )).alias (col_name )
4848 else :
4949 return col (col_name )
You can’t perform that action at this time.
0 commit comments