From a733e32345cd35f4b2cca38adb9b7fee190f557c Mon Sep 17 00:00:00 2001 From: "David H. Irving" Date: Fri, 13 Sep 2024 10:56:09 -0700 Subject: [PATCH] Stop using private Butler clone method Butler now has a public method equivalent to the private method that was being used in unit tests previously. --- tests/test_init_output_run.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/tests/test_init_output_run.py b/tests/test_init_output_run.py index 2cc6341e..e4c2dea2 100644 --- a/tests/test_init_output_run.py +++ b/tests/test_init_output_run.py @@ -205,7 +205,7 @@ def init_with_pipeline_graph_first( using the former to actually write init-outputs (with later attempts correctly failing or doing nothing, depending on parameters). """ - butler = butler._clone(run=run, collections=[self.INPUT_COLLECTION, run]) + butler = butler.clone(run=run, collections=[self.INPUT_COLLECTION, run]) pipeline_graph.init_output_run(butler) init_output_refs = self.find_init_output_refs(pipeline_graph, butler) # Build a QG with the init outputs already in place. @@ -264,7 +264,7 @@ def init_with_quantum_graph_first( using the latter to actually write init-outputs (with later attempts correctly failing or doing nothing, depending on parameters). """ - butler = butler._clone(run=run, collections=[self.INPUT_COLLECTION, run]) + butler = butler.clone(run=run, collections=[self.INPUT_COLLECTION, run]) # Build a QG. quantum_graph_builder = AllDimensionsQuantumGraphBuilder( pipeline_graph, @@ -316,7 +316,7 @@ def init_with_qbb_first(self, pipeline_graph: PipelineGraph, butler: Butler, run (with later attempts correctly failing or doing nothing, depending on parameters). """ - butler = butler._clone(run=run, collections=[self.INPUT_COLLECTION, run]) + butler = butler.clone(run=run, collections=[self.INPUT_COLLECTION, run]) # Build a QG. quantum_graph_builder = AllDimensionsQuantumGraphBuilder( pipeline_graph, @@ -512,7 +512,7 @@ def test_config_change(self) -> None: butler.put(DynamicTestPipelineTaskConfig(), "a_config", run="run1") with self.assertRaises(ConflictingDefinitionError): pipeline_graph.init_output_run( - butler._clone(run="run1", collections=[self.INPUT_COLLECTION, "run1"]) + butler.clone(run="run1", collections=[self.INPUT_COLLECTION, "run1"]) ) quantum_graph_builder = AllDimensionsQuantumGraphBuilder( pipeline_graph, @@ -526,7 +526,7 @@ def test_config_change(self) -> None: ) with self.assertRaises(ConflictingDefinitionError): quantum_graph.init_output_run( - butler._clone(run="run1", collections=[self.INPUT_COLLECTION, "run1"]) + butler.clone(run="run1", collections=[self.INPUT_COLLECTION, "run1"]) )