Simple refactor change. Signed-off-by: Jerry Ge <jerry.ge@arm.com>
NB_DOMAIN
add_mlir_python_extension
See https://mlir.llvm.org/ for more information.