Merge pull request #237 from interpretml/gaugup/AddPython3.10

Add tests and classifier for python 3.10

Author: gaugup

.github/workflows/pythonpackage.yml

@@ -15,7 +15,7 @@ jobs:
     runs-on: ${{ matrix.os }}
     strategy:
       matrix:
-        python-version: [3.6, 3.7, 3.8, 3.9]
+        python-version: [3.6, 3.7, 3.8, 3.9, "3.10"]
         os: [ubuntu-latest, macos-latest]
         exclude:
           - os: macos-latest

requirements-deeplearning.txt

@@ -1,2 +1,2 @@
 tensorflow>=1.13.0-rc1 
-torch
+torch; python_version < '3.10'

setup.py

@@ -38,6 +38,7 @@
         "Programming Language :: Python :: 3.7",
         "Programming Language :: Python :: 3.8",
         "Programming Language :: Python :: 3.9",
+        "Programming Language :: Python :: 3.10",
         "License :: OSI Approved :: MIT License",
         "Operating System :: OS Independent",
     ],

tests/test_notebooks.py

@@ -4,6 +4,7 @@
 
 import os
 import subprocess
+import sys
 import tempfile
 
 import nbformat
@@ -16,7 +17,11 @@
         "DiCE_with_advanced_options.ipynb",  # requires tensorflow 1.x
         "DiCE_getting_started_feasible.ipynb",  # needs changes after latest refactor
         "Benchmarking_different_CF_explanation_methods.ipynb"
-        ]
+]
+# notebooks that don't need to run on python 3.10
+torch_notebooks_not_3_10 = [
+    "DiCE_getting_started.ipynb"
+]
 
 # Adding the dice root folder to the python path so that jupyter notebooks
 if 'PYTHONPATH' not in os.environ:
@@ -77,6 +82,11 @@ def _notebook_run(filepath):
             nb,
             marks=[pytest.mark.skip, pytest.mark.advanced],
             id=nb)
+    elif sys.version_info >= (3, 10) and nb in torch_notebooks_not_3_10:
+        param = pytest.param(
+            nb,
+            marks=[pytest.mark.skip, pytest.mark.advanced],
+            id=nb)
     else:
         param = pytest.param(nb, id=nb)
     parameter_list.append(param)