Use pack for U2I training-test split wrapper