diff -r e88549a02f5e -r 33fe96a5c522 rust/hg-cpython/src/discovery.rs --- a/rust/hg-cpython/src/discovery.rs Tue Oct 15 22:02:34 2019 -0400 +++ b/rust/hg-cpython/src/discovery.rs Mon Sep 30 16:31:53 2019 -0400 @@ -13,9 +13,7 @@ //! `mercurial.setdiscovery.partialdiscovery`. use crate::{ - cindex::Index, - conversion::{py_set, rev_pyiter_collect}, - exceptions::GraphError, + cindex::Index, conversion::rev_pyiter_collect, exceptions::GraphError, }; use cpython::{ ObjectProtocol, PyDict, PyModule, PyObject, PyResult, PyTuple, Python, @@ -23,6 +21,7 @@ }; use hg::discovery::PartialDiscovery as CorePartialDiscovery; use hg::Revision; +use std::collections::HashSet; use std::cell::RefCell; @@ -106,12 +105,9 @@ Ok(as_dict) } - def commonheads(&self) -> PyResult { - py_set( - py, - &self.inner(py).borrow().common_heads() - .map_err(|e| GraphError::pynew(py, e))? - ) + def commonheads(&self) -> PyResult> { + self.inner(py).borrow().common_heads() + .map_err(|e| GraphError::pynew(py, e)) } def takefullsample(&self, _headrevs: PyObject,