Source code for semi_cr.core.lab.utils.general

def remove_class_keys(obj):
    if isinstance(obj, dict):
        return {
            key: remove_class_keys(value)
            for key, value in obj.items()
            if key != "__class__"
        }
    elif isinstance(obj, list):
        return [remove_class_keys(item) for item in obj]
    else:
        return obj