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?
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?