diff --git a/vm/src/stdlib/itertools.rs b/vm/src/stdlib/itertools.rs index ed02cc5a6..cc4eeb2d0 100644 --- a/vm/src/stdlib/itertools.rs +++ b/vm/src/stdlib/itertools.rs @@ -202,6 +202,13 @@ mod decl { fn reduce(zelf: PyRef) -> (PyTypeRef, (BigInt,)) { (zelf.class().clone(), (zelf.cur.read().clone(),)) } + + #[pymethod(magic)] + fn repr(&self) -> PyResult { + let cur = self.cur.read(); + + Ok(format!("count({})", cur)) + } } impl IterNextIterable for PyItertoolsCount {} impl IterNext for PyItertoolsCount {