try:
    from .prune_utils import get_prune_layer
except ImportError as e:
    print(e)