update PS shift correlation into TT and HH

models/common.py

@@ -20,9 +20,21 @@ class StatModelBase(Datacard):
             systematic="scale",
             processes=["DY", "TT", "VV", "WJets", "ggHH", "tHW", "tHq", "ttH", "ttW", "ttZ"],
         )
-        # Check decorrelation with Karl:
-        lines += self.decorrelate(systematic="ps_fsr", processes=["WJets", "TT", "ST"])
-        lines += self.decorrelate(systematic="ps_isr", processes=["WJets", "TT", "ST"])
+        # Decorrelation according to Karl (Skype) into TT and HH:
+        lines += self.decorrelate(
+            systematic="ps_fsr",
+            processes={
+                "TT": "TT",
+                **{p: "HH" for p in self.processes if p.startswith(("ggHH_", "qqHH_"))},
+            },
+        )
+        lines += self.decorrelate(
+            systematic="ps_isr",
+            processes={
+                "TT": "TT",
+                **{p: "HH" for p in self.processes if p.startswith(("ggHH_", "qqHH_"))},
+            },
+        )
         lines += [f"{self.bin} autoMCStats 10"]
         return lines