hgext/largefiles/overrides.py
changeset 46051 4b89cf08d8dc
parent 45957 89a2afe31e82
child 46362 32da58916fd0
child 46440 bb3a5c0df06b
--- a/hgext/largefiles/overrides.py	Tue Dec 01 09:13:08 2020 +0100
+++ b/hgext/largefiles/overrides.py	Tue Dec 01 15:11:06 2020 +0100
@@ -37,11 +37,14 @@
     scmutil,
     smartset,
     subrepo,
-    upgrade,
     url as urlmod,
     util,
 )
 
+from mercurial.upgrade_utils import (
+    actions as upgrade_actions,
+)
+
 from . import (
     lfcommands,
     lfutil,
@@ -1837,8 +1840,8 @@
     return result
 
 
-@eh.wrapfunction(upgrade, b'preservedrequirements')
-@eh.wrapfunction(upgrade, b'supporteddestrequirements')
+@eh.wrapfunction(upgrade_actions, b'preservedrequirements')
+@eh.wrapfunction(upgrade_actions, b'supporteddestrequirements')
 def upgraderequirements(orig, repo):
     reqs = orig(repo)
     if b'largefiles' in repo.requirements: