Pass WriteType, not str to RetryPolicy.on_write_timeout

Fixes #123