Handling already registered custom checks

[ReyJust]

👋🏼 , Hello.

I was wondering if there is any feature to deal with already registered custom checks? Unfortunately, my current environment "keeps" previous registerations.

For example, this snippet found at this link, if ran twice in a notebook cell:

import pandera as pa
import pandera.extensions as extensions
import pandas as pd

@extensions.register_check_method(statistics=["min_value", "max_value"])
def is_between(pandas_obj, *, min_value, max_value):
    return (min_value <= pandas_obj) & (pandas_obj <= max_value)

schema = pa.DataFrameSchema({
    "col": pa.Column(int, pa.Check.is_between(min_value=1, max_value=10))
})

will raise:

ValueError: method with name 'is_between' already defined. Check methods must have a unique method name.

• I couldn't found any argument to "overwrite" the custom check, if it already exist.
• Also I tried to search for a function to remove the custom check.
• Finally I tried list(pa.Check.REGISTERED_CUSTOM_CHECKS).remove('is_between') which of course, is the proper way to do I believe.
  Any idea how to overcome this please? Thanks! 🙏🏼

[matrs]

you can do

pa.Check.REGISTERED_CUSTOM_CHECKS.clear()

To clear all the registered methods.