--- a/mercurial/setdiscovery.py	Tue May 21 12:46:38 2019 +0200
+++ b/mercurial/setdiscovery.py	Wed Feb 20 09:04:54 2019 +0100
@@ -52,6 +52,7 @@
 )
 from . import (
     error,
+    policy,
     util,
 )
 
@@ -269,6 +270,10 @@
                 sample.update(takefrom[:more])
         return sample
 
+partialdiscovery = policy.importrust('discovery',
+                                     member='PartialDiscovery',
+                                     default=partialdiscovery)
+
 def findcommonheads(ui, local, remote,
                     initialsamplesize=100,
                     fullsamplesize=200,