Has this issue been resolved? I am using a Wrapped Catboost classifier, and sometimes I can use batch_score, and some times it gives me the same error:"org.apache.spark.SparkException: Job aborted due to stage failure: Task 2 in stage 2094.0 failed 4...