from .partition_kg import partition_kg