Switching to using ray.get instead of ray.wait

idaholab · Apr 12, 2022 · e76a2bc · e76a2bc
1 parent 5f4ba33
commit e76a2bc
Show file tree

Hide file tree

Showing 5 changed files with 15 additions and 5 deletions.
diff --git a/ravenframework/Runners/DistributedMemoryRunner.py b/ravenframework/Runners/DistributedMemoryRunner.py
@@ -69,7 +69,15 @@ def isDone(self):
     if self.thread is None:
       return True
     else:
-      return (self.thread in ray.wait([self.thread], timeout=waitTimeOut)[0]) if im.isLibAvail("ray") else self.thread.finished
+      if im.isLibAvail("ray"):
+        try:
+          ray.get(self.thread, timeout=waitTimeOut)
+          return True
+        except ray.exceptions.GetTimeoutError:
+          return False
+      else:
+        self.thread.finished
+      #return (self.thread in ray.wait([self.thread], timeout=waitTimeOut)[0]) if im.isLibAvail("ray") else self.thread.finished
 
   def _collectRunnerResponse(self):
     """

diff --git a/tests/cluster_tests/RavenRunsRaven/Code/Inner/Simple.py b/tests/cluster_tests/RavenRunsRaven/Code/Inner/Simple.py
@@ -14,6 +14,7 @@
 import sys
 import argparse
 import configparser
+import time
 
 def checkAux():
   """
@@ -70,7 +71,7 @@ def write(a, b, c, x, y, out):
     @ In, y, float, float
     @ In, out, string, filename to write results to
   """
-  print('Writing to', out)
+  print('Writing to', out, time.ctime())
   with open(out, 'w') as f:
     f.writelines(','.join('abcxy') + '\n')
     f.writelines(','.join(str(i) for i in [a, b, c, x, y]) + '\n')
@@ -82,3 +83,4 @@ def write(a, b, c, x, y, out):
   a, b, x, y, out = readInput(infileName)
   c = run(a, b, x, y)
   write(a, b, c, x, y, out)
+  print("Goodbye", time.ctime())
diff --git a/tests/cluster_tests/RavenRunsRaven/Code/inner.xml b/tests/cluster_tests/RavenRunsRaven/Code/inner.xml
@@ -1,5 +1,5 @@
 <?xml version="1.0" ?>
-<Simulation verbosity="debug">
+<Simulation verbosity="debug" profile="jobs">
 
   <RunInfo>
     <WorkingDir>Inner</WorkingDir>

diff --git a/tests/cluster_tests/RavenRunsRaven/code.xml b/tests/cluster_tests/RavenRunsRaven/code.xml
@@ -1,5 +1,5 @@
 <?xml version="1.0" ?>
-<Simulation verbosity="debug">
+<Simulation verbosity="debug" profile="jobs">
   <TestInfo>
     <name>cluster_tests/RavenRunsRaven.Code</name>
     <author>talbpaul</author>

diff --git a/tests/cluster_tests/RavenRunsRaven/rom.xml b/tests/cluster_tests/RavenRunsRaven/rom.xml
@@ -1,5 +1,5 @@
 <?xml version="1.0" ?>
-<Simulation verbosity="debug">
+<Simulation verbosity="debug" profile="jobs">
   <TestInfo>
     <name>framework/cluster_tests/RavenRunsRaven.ROM</name>
     <author>alfoa</author>