Surrogate compliance modeling enables reinforcement learned locomotion gaits for soft robots