HeartSaVioR commented on code in PR #53122:
URL: https://github.com/apache/spark/pull/53122#discussion_r2594456672


##########
python/pyspark/sql/pandas/serializers.py:
##########
@@ -2009,20 +2009,21 @@ def row_stream():
                         for i, c in 
enumerate(flatten_state_table.itercolumns())
                     ]
 
+                    flatten_init_table = flatten_columns(batch, "initState")
+                    init_data_pandas = [
+                        self.arrow_to_pandas(c, i)
+                        for i, c in enumerate(flatten_init_table.itercolumns())
+                    ]
+
                     if bool(data_pandas):

Review Comment:
   While I think XOR is still representing the criteria better (it's implicit 
that we wouldn't have a case where neither inputData nor initState have data, 
but while we are here), it's not a big deal and it's OK to go with implicit 
assumption. Looks OK to me.



-- 
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.

To unsubscribe, e-mail: [email protected]

For queries about this service, please contact Infrastructure at:
[email protected]


---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]

Reply via email to