diff --git a/vm/src/stdlib/itertools.rs b/vm/src/stdlib/itertools.rs index bd4de3ce0a4..cad3dca66e2 100644 --- a/vm/src/stdlib/itertools.rs +++ b/vm/src/stdlib/itertools.rs @@ -193,7 +193,15 @@ mod decl { } #[pyimpl(with(IterNext, Constructor))] - impl PyItertoolsCount {} + impl PyItertoolsCount { + // TODO: Implement this + // if (lz->cnt == PY_SSIZE_T_MAX) + // return Py_BuildValue("0(00)", Py_TYPE(lz), lz->long_cnt, lz->long_step); + #[pymethod(magic)] + fn reduce(zelf: PyRef) -> (PyTypeRef, (BigInt,)) { + (zelf.class().clone(), (zelf.cur.read().clone(),)) + } + } impl IterNextIterable for PyItertoolsCount {} impl IterNext for PyItertoolsCount { fn next(zelf: &Py, vm: &VirtualMachine) -> PyResult {