Fix calling get on the future

nv-morpheus · Aug 16, 2024 · e865253 · e865253
1 parent ac1c28d
commit e865253
Showing 1 changed file with 7 additions and 5 deletions.
diff --git a/python/mrc/_pymrc/src/segment.cpp b/python/mrc/_pymrc/src/segment.cpp
@@ -150,15 +150,17 @@ class PyIteratorIterator
         while (subscriber.is_subscribed() && future_status != std::future_status::ready)
         {
             future_status = future.wait_for(std::chrono::milliseconds(100));
-            if (future_status == std::future_status::ready)
-            {
-                future.get();
-            }
-            else
+
+            if (future_status != std::future_status::ready)
             {
                 std::this_thread::yield();
             }
         }
+
+        if (future_status == std::future_status::ready)
+        {
+            future.get();
+        }
     }
 
   private: