Fixes the wrapper for sampler
Fixes the wrapper for sampler

sky-xu/deepdive,HazyResearch/deepdive,shahin/deepdive,shahin/deepdive,sky-xu/deepdive,HazyResearch/deepdive,HazyResearch/deepdive,HazyResearch/deepdive