Custom sharding funcs support