Import extension traitlet in top-level nbconvert