Merge pull request #16 from aviolante/develop

Update SimpleQueue() to Queue()
project-codeflare · Jul 9, 2021 · b5cd21e · b5cd21e
2 parents be22000 + fdef6dd
commit b5cd21e
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/codeflare/pipelines/Runtime.py b/codeflare/pipelines/Runtime.py
@@ -46,7 +46,7 @@
 from sklearn.model_selection import BaseCrossValidator
 from enum import Enum
 
-from queue import SimpleQueue
+from queue import Queue
 import pandas as pd
 
 
@@ -374,7 +374,7 @@ def select_pipeline(pipeline_output: dm.PipelineOutput, chosen_xyref: dm.XYRef)
     :return: Selected pipeline
     """
     pipeline = dm.Pipeline()
-    xyref_queue = SimpleQueue()
+    xyref_queue = Queue()
 
     xyref_queue.put(chosen_xyref)
     while not xyref_queue.empty():
@@ -411,7 +411,7 @@ def get_pipeline_input(pipeline: dm.Pipeline, pipeline_output: dm.PipelineOutput
     """
     pipeline_input = dm.PipelineInput()
 
-    xyref_queue = SimpleQueue()
+    xyref_queue = Queue()
     xyref_queue.put(chosen_xyref)
     while not xyref_queue.empty():
         curr_xyref = xyref_queue.get()