From 33fc39bf6265eac2713f704efc540c4f04d3a3ac Mon Sep 17 00:00:00 2001
From: Toke Stuart Jepsen <tokejepsen@gmail.com>
Date: Fri, 21 Aug 2020 14:38:10 +0100
Subject: [PATCH] Update subset families on integration

---
 pype/plugins/global/publish/integrate_new.py | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/pype/plugins/global/publish/integrate_new.py b/pype/plugins/global/publish/integrate_new.py
index a3c2ffe52b..cc106ad8a2 100644
--- a/pype/plugins/global/publish/integrate_new.py
+++ b/pype/plugins/global/publish/integrate_new.py
@@ -680,6 +680,12 @@ class IntegrateAssetNew(pyblish.api.InstancePlugin):
                 instance.data.get('subsetGroup')}}
             )
 
+        # Update families on subset.
+        io.update_many(
+            {"type": "subset", "_id": io.ObjectId(subset["_id"])},
+            {"$set": {"data.families": instance.data.get("families", [])}}
+        )
+
         return subset
 
     def create_version(self, subset, version_number, data=None):