diff --git a/vm/src/stdlib/operator.rs b/vm/src/stdlib/operator.rs index 71150963d..12d0f62da 100644 --- a/vm/src/stdlib/operator.rs +++ b/vm/src/stdlib/operator.rs @@ -1,3 +1,4 @@ +use crate::builtins::int::PyIntRef; use crate::builtins::pystr::PyStrRef; use crate::byteslike::PyBytesLike; use crate::common::cmp; @@ -43,10 +44,15 @@ fn _operator_compare_digest( Ok(res) } +fn _operator_index(obj: PyObjectRef, vm: &VirtualMachine) -> PyResult { + vm.to_index(&obj) +} + pub fn make_module(vm: &VirtualMachine) -> PyObjectRef { let ctx = &vm.ctx; py_module!(vm, "_operator", { "length_hint" => named_function!(ctx, _operator, length_hint), "_compare_digest" => named_function!(ctx, _operator, compare_digest), + "index" => named_function!(ctx, _operator, index), }) }