Merge pull request #85 from kmpaul/remove-check-python-3

Remove check python 3 and update CI

.github/workflows/tests.yml

@@ -19,7 +19,7 @@ jobs:
     strategy:
       fail-fast: false
       matrix:
-        python: ['3.6', '3.7', '3.8']
+        python: ['3.8', '3.9', '3.10']
         mpi: ['mpich', 'openmpi']
     env:
       PYTHON: ${{ matrix.python }}

.pre-commit-config.yaml

@@ -2,18 +2,18 @@ exclude: 'docs/source/conf.py|versioneer.py'
 
 repos:
   - repo: https://github.com/ambv/black
-    rev: 20.8b1
+    rev: 22.3.0
     hooks:
     - id: black
   - repo: https://gitlab.com/pycqa/flake8
-    rev: 3.8.4
+    rev: 3.9.2
     hooks:
     - id: flake8
   - repo: https://github.com/asottile/seed-isort-config
-    rev: v1.9.3
+    rev: v2.2.0
     hooks:
       - id: seed-isort-config
   - repo: https://github.com/pre-commit/mirrors-isort
-    rev: v4.3.21
+    rev: v5.10.1
     hooks:
       - id: isort

dask_mpi/cli.py

@@ -2,8 +2,7 @@
 import json
 
 import click
-from dask.distributed import Scheduler, Worker
-from distributed.cli.utils import check_python_3
+from distributed import Scheduler, Worker
 from distributed.utils import import_term
 from mpi4py import MPI
 
@@ -148,10 +147,5 @@ async def run_worker():
         asyncio.get_event_loop().run_until_complete(run_worker())
 
 
-def go():
-    check_python_3()
-    main()
-
-
 if __name__ == "__main__":
-    go()
+    main()

dask_mpi/core.py

@@ -3,7 +3,7 @@
 import sys
 
 import dask
-from dask.distributed import Client, Nanny, Scheduler
+from distributed import Client, Nanny, Scheduler
 from distributed.utils import import_term
 from tornado import gen
 from tornado.ioloop import IOLoop

setup.py

@@ -52,7 +52,7 @@ def environment_dependencies(obj, dependencies=None):
     long_description=long_description,
     entry_points="""
             [console_scripts]
-            dask-mpi=dask_mpi.cli:go
+            dask-mpi=dask_mpi.cli:main
             """,
     zip_safe=False,
 )