From 78e0724d287caedcde352abadba2da0828ea84f1 Mon Sep 17 00:00:00 2001 From: VanyaBelyaev Date: Sun, 11 Feb 2024 20:12:10 +0100 Subject: [PATCH] Update simfit.py --- ostap/fitting/simfit.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/ostap/fitting/simfit.py b/ostap/fitting/simfit.py index 5f5d65c5..db978c5d 100644 --- a/ostap/fitting/simfit.py +++ b/ostap/fitting/simfit.py @@ -1213,7 +1213,9 @@ def generate ( self , cargs = tuple ( cargs ) pdf = self.categories [ label ] - vv = ROOT.RooArgSet ( [ v for v in self.vars if v in varset ] ) + vv = ROOT.RooArgSet () + for v for v in self.vars : + if v in varset : vv.add ( v ) ds = pdf.generate ( nevts , varset = vv , ## varset = varset ,