Merge pull request #601 from nipype/ci/test-python-3.11

FIX: Enable testing with Python 3.11

Author: effigies

.github/workflows/testpydra.yml

@@ -16,7 +16,10 @@ jobs:
     strategy:
       matrix:
         os: [macos-latest, ubuntu-latest, windows-latest]
-        python-version: [3.7, 3.8, 3.9, "3.10"]
+        python-version: ['3.7', '3.8', '3.9', '3.10', '3.11']
+        exclude:
+          - os: 'windows-latest'
+            python-version: '3.11'
       fail-fast: false
     runs-on: ${{ matrix.os }}
 
@@ -30,7 +33,9 @@ jobs:
       with:
         python-version: ${{ matrix.python-version }}
     - name: Update build tools
-      run: python -m pip install --upgrade pip build
+      run: |
+        python -m pip install --upgrade pip
+        python -m pip install build wheel
 
     - name: Build pydra
       run: python -m build

pydra/engine/workers.py

@@ -58,7 +58,11 @@ async def fetch_finished(self, futures):
         done = set()
         try:
             done, pending = await asyncio.wait(
-                futures, return_when=asyncio.FIRST_COMPLETED
+                [
+                    asyncio.create_task(f) if not isinstance(f, asyncio.Task) else f
+                    for f in futures
+                ],
+                return_when=asyncio.FIRST_COMPLETED,
             )
         except ValueError:
             # nothing pending!
@@ -105,7 +109,11 @@ async def fetch_finished(self, futures):
         try:
             self._jobs += len(futures)
             done, pending = await asyncio.wait(
-                futures, return_when=asyncio.FIRST_COMPLETED
+                [
+                    asyncio.create_task(f) if not isinstance(f, asyncio.Task) else f
+                    for f in futures
+                ],
+                return_when=asyncio.FIRST_COMPLETED,
             )
         except ValueError:
             # nothing pending!