trtStepWedge function

Assign treatment for stepped-wedge design