Vito would have bought whomever was for sale. More Dems than Republicans in his era in NY, but Vito would never have taken a visible stake in either party. And neither party would want to feature Vito as one of their supporters.
Michael and Tom probably would have gravitated to the GOP in Nevada as part of their effort to be publicly "respectable." However, Sen. Pat McCarran, on whom Sen. Pat Geary was modeled, was a Democrat. Bugsy Siegel bought him, and so did Moe Dalitz.