From 45a2c210f47d5d3ac30ea9c3d77b36ea035adc44 Mon Sep 17 00:00:00 2001
From: VanyaBelyaev <Ivan.Belyaev@cern.ch>
Date: Wed, 24 Jul 2024 19:37:13 +0200
Subject: [PATCH] Update dataset.py

---
 ostap/fitting/dataset.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/ostap/fitting/dataset.py b/ostap/fitting/dataset.py
index 84d8abf4..7d412f77 100644
--- a/ostap/fitting/dataset.py
+++ b/ostap/fitting/dataset.py
@@ -37,7 +37,8 @@
                                           list_types    , sequence_types )
 from   ostap.math.base           import islong
 from   ostap.fitting.variables   import valid_formula, make_formula 
-from   ostap.trees.cuts          import expression_types, vars_and_cuts  
+from   ostap.trees.cuts          import expression_types, vars_and_cuts 
+import ostap.trees.trees
 import ostap.fitting.roocollections
 import ostap.fitting.printable
 import ROOT, random, math, sys, ctypes