A type rating training course should not be done under a B-class visitor visa. This point was clarified some years ago by the US DOJ in the memo copied below. Instead use an M-1 visa. In some circumstances a course of study may be pursued while admitted under a different visa class. See SEVP table in the document
Nonimmigrants: Who Can Study?
PDF copy of the memo.