Answer from Java side is empty

This looks like the JVM is running out of memory. Can you try setting the PYSPARK_SUBMIT_ARGS env var as in BGEN filter_cols memory leak?