Add non-local imports for joblib (temporary)