Python 3.11 (#749)

Co-authored-by: Masahiro Hiramori <mhg00g13@gmail.com>

.github/workflows/pythonapp.yml

@@ -22,7 +22,7 @@ jobs:
     strategy:
       matrix:
         os: [ubuntu-22.04, macos-12, windows-2022]
-        python-version: ['3.8', '3.9', '3.10']
+        python-version: ['3.8', '3.9', '3.10', '3.11']
 
     steps:
     - uses: actions/checkout@v3
@@ -81,12 +81,12 @@ jobs:
       run: |
         python -m pip install .[extra,onnx,sparkml]
         python -m pip install pandas
-    - name: Install TVM from pypi if Ubuntu
-      if: ${{ startsWith(matrix.os, 'ubuntu') }}
+    - name: Install TVM from pypi if Ubuntu and not python 3.11
+      if: ${{ startsWith(matrix.os, 'ubuntu') && matrix.python-version != '3.11' }}
       run: python -m pip install apache-tvm==0.10.0
     - uses: actions/cache@v3
       # TVM takes forever, we try to cache it.
-      if: ${{ startsWith(matrix.os, 'macos')}}
+      if: ${{ startsWith(matrix.os, 'macos') && matrix.python-version != '3.11' }}
       id: cache
       env:
         CACHE_NUMBER: 9
@@ -97,11 +97,11 @@ jobs:
     # 2 to 4 will be retrieved from the cache.
     # The pipeline only works for Unix systems. For windows we will have to compile LLVM from source which is a no go.
     - name: Install LLVM if Mac
-      if: ${{ startsWith(matrix.os, 'macos') }}
+      if: ${{ startsWith(matrix.os, 'macos') && matrix.python-version != '3.11'}}
       run: |
         brew install llvm@14
     - name: Fetch and prepare TVM for compilation if Mac
-      if: ${{ steps.cache.outputs.cache-hit != 'true' && startsWith(matrix.os, 'macos') }}
+      if: ${{ steps.cache.outputs.cache-hit != 'true' && startsWith(matrix.os, 'macos') && matrix.python-version != '3.11' }}
       run: |
         cd ~/work/hummingbird
         git clone https://github.com/apache/tvm.git
@@ -110,7 +110,7 @@ jobs:
         git submodule update --recursive --init
         cmake -E make_directory build
     - name: CMake TVM if Mac
-      if: ${{ steps.cache.outputs.cache-hit != 'true' && startsWith(matrix.os, 'macos') }}
+      if: ${{ steps.cache.outputs.cache-hit != 'true' && startsWith(matrix.os, 'macos') && matrix.python-version != '3.11' }}
       working-directory: ../tvm/build
       run: >-
         MACOSX_DEPLOYMENT_TARGET=10.13 cmake
@@ -119,12 +119,12 @@ jobs:
         "-DUSE_LLVM=$(brew --prefix llvm@14)/bin/llvm-config --link-static"
         ..
     - name: Build TVM if Mac
-      if: ${{ steps.cache.outputs.cache-hit != 'true' && startsWith(matrix.os, 'macos') }}
+      if: ${{ steps.cache.outputs.cache-hit != 'true' && startsWith(matrix.os, 'macos') && matrix.python-version != '3.11' }}
       working-directory: ../tvm/build
       run: |
         make -j3
-    - name: Install python TVM if Mac
-      if: ${{  startsWith(matrix.os, 'macos') }}
+    - name: Install python TVM if Mac and not python 3.11
+      if: ${{  startsWith(matrix.os, 'macos') && matrix.python-version != '3.11' }}
       working-directory: ../tvm/python
       run: |
         python -m pip install -U wheel packaging

README.md

@@ -2,7 +2,7 @@
 
 [![PyPI version](https://badge.fury.io/py/hummingbird-ml.svg)](https://badge.fury.io/py/hummingbird-ml)
 [![](https://github.com/microsoft/hummingbird/workflows/Build/badge.svg?branch=main)](https://github.com/microsoft/hummingbird/actions)
-![](https://img.shields.io/badge/python-3.8%20%7C%203.9%20%7C%203.10-blue)
+![](https://img.shields.io/badge/python-3.8%20%7C%203.9%20%7C%203.10%20%7C%203.11-blue)
 [![coverage](https://codecov.io/gh/microsoft/hummingbird/branch/main/graph/badge.svg)](https://codecov.io/github/microsoft/hummingbird?branch=main)
 [![Gitter](https://badges.gitter.im/hummingbird-ml/community.svg)](https://gitter.im/hummingbird-ml/community?utm_source=badge&utm_medium=badge&utm_campaign=pr-badge)
 [![Downloads](https://pepy.tech/badge/hummingbird-ml)](https://pepy.tech/project/hummingbird-ml)
@@ -64,7 +64,7 @@ _Thank you to [Chien Vu](https://www.linkedin.com/in/vumichien/) for contributin
 
 ## Installation
 
-Hummingbird was tested on Python 3.8, 3.9 and 3.10 on Linux, Windows and MacOS machines.  It is recommended to use a virtual environment (See: [python3 venv doc](https://docs.python.org/3/tutorial/venv.html) or [Using Python environments in VS Code](https://code.visualstudio.com/docs/python/environments).)
+Hummingbird was tested on Python 3.8, 3.9, 3.10 and 3.11 on Linux, Windows and MacOS machines. (TVM only works through Python3.10.) It is recommended to use a virtual environment (See: [python3 venv doc](https://docs.python.org/3/tutorial/venv.html) or [Using Python environments in VS Code](https://code.visualstudio.com/docs/python/environments).)
 
 Hummingbird requires PyTorch >= 1.6.0. Please go [here](https://pytorch.org/) for instructions on how to install PyTorch based on your platform and hardware.
 

setup.py

@@ -39,7 +39,7 @@ extra_requires = [
     "xgboost>=0.90,<2.0.0",
     "lightgbm>=2.2",
     "holidays==0.24",
-    "prophet==1.1",
+    "prophet",
 ]
 setup(
     name="hummingbird-ml",