How to set a custom name for сustom сheck method ?

[karutyunov]

Hello!

Using custom checks, I couldn't find how to define a custom name to check. I want something other than null or None to be displayed in the error log

from pandera.extensions import register_check_method

@register_check_method(statistics=["str_length"], check_type="element_wise")
def test_pyspark_check(pyspark_obj, *, str_length: int) -> bool:

    check_df = pyspark_obj.dataframe.withColumn("length", F.length(F.col(pyspark_obj.column_name)))
    return check_df.select("length").limit(100).collect()[0][0] == str_length

class PartitionSchema(DataFrameModel):
    full_partition_name: T.StringType() = pa.Field(nullable=False)
    probe: T.StringType() = pa.Field(nullable=True, test_pyspark_check={
                "str_length": 1
            })
    event_date: T.StringType() = pa.Field(nullable=True, test_pyspark_check={
                "str_length": 9
            })
    processing_time: T.StringType() = pa.Field(nullable=True)

Here you can see that I am not getting any useful information about the name of the check or error

{
    "DATA": {
        "DATAFRAME_CHECK": [
            {
                "schema": "PartitionSchema",
                "column": "probe",
                "check": null,
                "error": "column 'probe' with type StringType failed validation None"
            },
            {
                "schema": "PartitionSchema",
                "column": "event_date",
                "check": null,
                "error": "column 'event_date' with type StringType failed validation None"
            }
        ]
    }
}

Is there any way to set a custom name for the check?