I don't see anything in the law that would carve out compensation for either a sole proprietorship or a partnership. Neither can pay wages to their owners though partnerships CAN pay guaranteed payments. The term "reasonable compensation" is used only in the context of what is included in qualified business income. This is very clear for an S corporation.
Ironically for taxpayers over the threshold, S corps need W2 wages (assuming the 2.5% of depreciable property is of little benefit) to get the deduction. It actually makes sense to increase the S corp wages to the shareholder/employee in some cases as the deduction will save more tax than the wages create in payroll taxes once the payroll taxes include only medicare. for an S corp with income of $700,000 I worked out a spreadsheet and figured somewhere around 28% of income paid as wages is the crossover.