Update simfit.py

OstapHEP · Feb 11, 2024 · 78e0724 · 78e0724
1 parent 121ea7e
commit 78e0724
Showing 1 changed file with 3 additions and 1 deletion.
diff --git 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   ,