bugfix/celery-chord-error

tests/integration/definitions.py

@@ -41,6 +41,7 @@
 """
 
 import shutil
+import sys
 
 # Pylint complains that we didn't install this module but it's defined locally so ignore
 from conditions import (  # pylint: disable=E0401
@@ -63,6 +64,24 @@
 KILL_WORKERS = "pkill -9 -f '.*merlin_test_worker'"
 
 
+def is_python_version_above_37() -> bool:
+    """
+    Check if the python version is above or below 3.7.
+
+    :returns: True if python version is 3.8+. False otherwise.
+    """
+    if sys.version_info.major > 3:
+        return True
+    if sys.version_info.major < 3:
+        return False
+
+    # If we're here then the major version is 3
+    if sys.version_info.minor >= 8:
+        return True
+
+    return False
+
+
 def get_worker_by_cmd(cmd: str, default: str) -> str:
     """
     Given a command used by a scheduler (e.g. flux for flux, jsrun for lsf, etc.)
@@ -832,7 +851,8 @@ def define_tests():  # pylint: disable=R0914,R0915
         "check chord error continues wf": {
             "cmds": [
                 f"{workers} {chord_err_wf} --vars OUTPUT_PATH=./{OUTPUT_DIR}",
-                f"{run} {chord_err_wf} --vars OUTPUT_PATH=./{OUTPUT_DIR}; sleep 10",
+                f"""{run} {chord_err_wf} --vars OUTPUT_PATH=./{OUTPUT_DIR};
+                sleep {10 if is_python_version_above_37() else 30}; tree {OUTPUT_DIR}""",
             ],
             "conditions": [
                 HasReturnCode(),