[DF] Test using the RDF's RunGraphs for DistRDF too

root-project · Jul 14, 2023 · c4ba60d · c4ba60d
1 parent aa257e9
commit c4ba60d
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 6 deletions.
diff --git a/python/distrdf/common/check_rungraphs.py b/python/distrdf/common/check_rungraphs.py
@@ -4,7 +4,6 @@
 
 import ROOT
 
-import DistRDF
 from DistRDF.Backends import Spark
 from DistRDF.Backends import Dask
 
@@ -46,7 +45,7 @@ def test_rungraphs_sparkanddask_3histos(self, connection):
         for proxy in histoproxies:
             assert proxy.proxied_node.value is None
 
-        DistRDF.RunGraphs(histoproxies)
+        ROOT.RDF.RunGraphs(histoproxies)
 
         # After RunGraphs all histograms are correctly assigned to the
         # node objects

diff --git a/python/distrdf/dask/check_rungraphs.py b/python/distrdf/dask/check_rungraphs.py
@@ -4,7 +4,6 @@
 
 import ROOT
 
-import DistRDF
 from DistRDF.Backends import Dask
 
 
@@ -36,7 +35,7 @@ def test_rungraphs_dask_3histos(self, connection):
         for proxy in histoproxies:
             assert proxy.proxied_node.value is None
 
-        DistRDF.RunGraphs(histoproxies)
+        ROOT.RDF.RunGraphs(histoproxies)
 
         # After RunGraphs all histograms are correctly assigned to the
         # node objects

diff --git a/python/distrdf/spark/check_rungraphs.py b/python/distrdf/spark/check_rungraphs.py
@@ -4,7 +4,6 @@
 
 import ROOT
 
-import DistRDF
 from DistRDF.Backends import Spark
 
 
@@ -36,7 +35,7 @@ def test_rungraphs_spark_3histos(self, connection):
         for proxy in histoproxies:
             assert proxy.proxied_node.value is None
 
-        DistRDF.RunGraphs(histoproxies)
+        ROOT.RDF.RunGraphs(histoproxies)
 
         # After RunGraphs all histograms are correctly assigned to the
         # node objects