1. Improve ds2numpy code and test

1. small fixes for `ostap.utuls.split_ranges` 1. add conversion to int for `RooAbsCategory` 1. add iterator/contains/len functions for `RooAbsDataStore`
OstapHEP · Dec 5, 2023 · 8971914 · 8971914
1 parent 1be4419
commit 8971914
Show file tree

Hide file tree

Showing 6 changed files with 207 additions and 247 deletions.
diff --git a/ReleaseNotes/release_notes.md b/ReleaseNotes/release_notes.md
@@ -8,6 +8,10 @@
   1. add test  for `ostap.stats.ustat` module 
   1. Add `Ostap::Math::two_samples` function 
   1. Add the first version of code for RooDataSety -> numpy conversion by Artem Egorychev 
+  1. Improve `ds2numpy` code and test
+  1. small fixes for `ostap.utuls.split_ranges`
+  1. add conversion to int for `RooAbsCategory` 
+  1. add iterator/contains/len functions for `RooAbsDataStore`
 
 
 ## Backward incompatible: 

diff --git a/ostap/fitting/dataset.py b/ostap/fitting/dataset.py
@@ -819,6 +819,16 @@ def _rds_make_unique_ ( dataset          ,
    ]
 
 
+ROOT.RooAbsDataStore . __len__      = lambda s : s.numEntries()
+ROOT.RooAbsDataStore . __iter__     = _rad_iter_ 
+ROOT.RooAbsDataStore . __contains__ = _rad_contains_ 
+
+
+_new_methods_ += [
+   ROOT.RooAbsDataStore . __len__       ,
+   ROOT.RooAbsDataStore . __iter__      , 
+   ROOT.RooAbsDataStore . __contains__
+   ]
 # =============================================================================
 ## Helper project method for RooDataSet/DataFrame/... and similar objects 
 #