Add BDA option to direct predict