diff --git a/infinitepush/__init__.py b/infinitepush/__init__.py --- a/infinitepush/__init__.py +++ b/infinitepush/__init__.py @@ -459,6 +459,16 @@ cgpart = bundle2.bundlepart('changegroup', data=cgstream) cgpart.addparam('version', version) parts.append(cgpart) + + try: + treemod = extensions.find('treemanifest') + except KeyError: + pass + else: + treepart = treemod.createtreepackpart(bundlerepo, outgoing, + treemod.TREEGROUP_PARTTYPE2) + parts.append(treepart) + return parts def _getbundleroots(oldrepo, bundlerepo, bundlerevs):