mercurial/dagutil.py

--- a/mercurial/dagutil.py	Thu Dec 07 01:53:14 2017 +0100
+++ b/mercurial/dagutil.py	Wed Dec 06 22:44:51 2017 +0100
@@ -154,8 +154,9 @@
 class revlogdag(revlogbaseddag):
     '''dag interface to a revlog'''
 
-    def __init__(self, revlog):
+    def __init__(self, revlog, localsubset=None):
         revlogbaseddag.__init__(self, revlog, set(revlog))
+        self._heads = localsubset
 
     def _getheads(self):
         return [r for r in self._revlog.headrevs() if r != nullrev]