diff --git a/vm/src/stdlib/re.rs b/vm/src/stdlib/re.rs index 1f7dbe389f..94ea2c27b5 100644 --- a/vm/src/stdlib/re.rs +++ b/vm/src/stdlib/re.rs @@ -122,17 +122,13 @@ fn re_sub( pattern: PyStringRef, repl: PyStringRef, string: PyStringRef, - count: OptionalArg, + count: OptionalArg, flags: OptionalArg, vm: &VirtualMachine, ) -> PyResult { let flags = extract_flags(flags); let regex = make_regex(vm, pattern.as_str(), flags)?; - let limit = count - .into_option() - .map(|i| usize::try_from_object(vm, i.into_object())) - .transpose()? - .unwrap_or(0); + let limit = count.unwrap_or(0); do_sub(vm, ®ex, repl, string, limit) }