test github action

Author: statmlben

.github/workflows/wheels.yml

@@ -0,0 +1,31 @@
+name: Build
+
+on: [push, pull_request]
+
+jobs:
+  build_wheels:
+    name: Build wheels on ${{ matrix.os }}
+    runs-on: ${{ matrix.os }}
+    strategy:
+      matrix:
+        os: [ubuntu-20.04, windows-2019, macOS-11]
+
+    steps:
+      - uses: actions/checkout@v4
+
+      # Used to host cibuildwheel
+      - uses: actions/setup-python@v3
+
+      - name: Install cibuildwheel
+        run: python -m pip install cibuildwheel==2.16.5
+
+      - name: Build wheels
+        run: python -m cibuildwheel --output-dir wheelhouse
+        # to supply options, put them in 'env', like:
+        # env:
+        #   CIBW_SOME_OPTION: value
+
+      - uses: actions/upload-artifact@v4
+        with:
+          name: cibw-wheels-${{ matrix.os }}-${{ strategy.job-index }}
+          path: ./wheelhouse/*.whl

build.py

@@ -0,0 +1,85 @@
+import os
+from pathlib import Path
+import zipfile
+import requests
+from pybind11.setup_helpers import Pybind11Extension, build_ext
+# from setuptools.command.build_ext import build_ext
+
+__version__ = "0.0.3"
+
+# The main interface is through Pybind11Extension.
+# * You can add cxx_std=11/14/17, and then build_ext can be removed.
+# * You can set include_pybind11=false to add the include directory yourself,
+#   say from a submodule.
+#
+# Note:
+#   Sort input source files if you glob sources to ensure bit-for-bit
+#   reproducible builds (https://github.com/pybind/python_example/pull/53)
+
+SETUP_DIRECTORY = Path(__file__).resolve().parent
+
+# Download Eigen source files
+# Modified from https://github.com/tohtsky/irspack/blob/main/setup.py
+class get_eigen_include(object):
+    EIGEN3_URL = "https://gitlab.com/libeigen/eigen/-/archive/3.4.0/eigen-3.4.0.zip"
+    EIGEN3_DIRNAME = "eigen-3.4.0"
+
+    def __str__(self) -> str:
+        # Test whether the environment variable EIGEN3_INCLUDE_DIR is set
+        # If yes, directly return this directory
+        eigen_include_dir = os.environ.get("EIGEN3_INCLUDE_DIR", None)
+        if eigen_include_dir is not None:
+            return eigen_include_dir
+
+        # If the directory already exists (e.g. from previous setup),
+        # directly return it
+        target_dir = SETUP_DIRECTORY / self.EIGEN3_DIRNAME
+        if target_dir.exists():
+            return target_dir.name
+
+        # Filename for the downloaded Eigen source package
+        download_target_dir = SETUP_DIRECTORY / "eigen3.zip"
+        response = requests.get(self.EIGEN3_URL, stream=True)
+        with download_target_dir.open("wb") as ofs:
+            for chunk in response.iter_content(chunk_size=1024):
+                ofs.write(chunk)
+        # Unzip package
+        with zipfile.ZipFile(download_target_dir) as ifs:
+            ifs.extractall()
+
+        return target_dir.name
+
+ext_modules = [
+    Pybind11Extension("rehline._internal",
+        sources=["src/rehline.cpp"],
+        include_dirs=[get_eigen_include()],
+        # Example: passing in the version to the compiled code
+        # define_macros=[('VERSION_INFO', __version__)],
+        ),
+]
+
+# class BuildFailed(Exception):
+#     pass
+
+# class ExtBuilder(build_ext):
+
+#     def run(self):
+#         try:
+#             build_ext.run(self)
+#         except (DistutilsPlatformError, FileNotFoundError):
+#             raise BuildFailed('File not found. Could not compile C extension.')
+
+#     def build_extension(self, ext):
+#         try:
+#             build_ext.build_extension(self, ext)
+#         except (CCompilerError, DistutilsExecError, DistutilsPlatformError, ValueError):
+#             raise BuildFailed('Could not compile C extension.')
+
+
+def build(setup_kwargs):
+    """
+    This function is mandatory in order to build the extensions.
+    """
+    setup_kwargs.update(
+        {"ext_modules": ext_modules, "cmd_class": {"build_ext": build_ext}, "zip_safe": False}
+    )

dist/rehline-0.0.3-cp310-cp310-linux_x86_64.whl

@@ -0,0 +1,30 @@
+[tool.poetry]
+name = "rehline"
+version = "0.0.3"
+description = "Regularized Composite ReLU-ReHU Loss Minimization with Linear Computation and Linear Convergence"
+authors = ["Ben Dai <bendai@cuhk.edu.hk>", "Yixuan Qiu <yixuanq@gmail.com>"]
+maintainers = ["Ben Dai <bendai@cuhk.edu.hk>"]
+license = "MIT"
+readme = "README.md"
+homepage = "https://rehline.github.io/"
+repository = "https://github.com/softmin/ReHLine-python"
+documentation = "https://rehline-python.readthedocs.io/en/latest/index.html"
+include = [
+    {path = "rehline/*.so", format = "wheel"}
+]
+
+[tool.poetry.build]
+script = "build.py"
+generate-setup-file = false
+
+[tool.poetry.dependencies]
+python = "^3.10"
+requests = "^2.31.0"
+pybind11 = "^2.11.1"
+numpy = "^1.23.5"
+scipy = "^1.11.4"
+scikit-learn = "^1.2.2"
+
+[build-system]
+requires = ["poetry-core>=1.0.0", "requests ~= 2.31.0", "pybind11 ~= 2.11.1", "setuptools ~= 59.0"]
+build-backend = "poetry.core.masonry.api" 

dist/rehline-0.0.3-cp310-cp310-manylinux_2_35_x86_64.whl

@@ -53,7 +53,7 @@ def __str__(self) -> str:
 ext_modules = [
     Pybind11Extension("rehline._internal",
         ["src/rehline.cpp"],
-        include_dirs=[get_eigen_include(), "src/rehline.h"],
+        include_dirs=[get_eigen_include()],
         # Example: passing in the version to the compiled code
         define_macros=[('VERSION_INFO', __version__)],
         ),
@@ -67,7 +67,7 @@ def __str__(self) -> str:
     url="https://github.com/softmin/ReHLine-python",
     description="Regularized Composite ReLU-ReHU Loss Minimization with Linear Computation and Linear Convergence",
     packages=["rehline"],
-    install_requires=["requests", "pybind11", "numpy", "scipy"],
+    install_requires=["requests", "pybind11", "numpy", "scipy", "scikit-learn"],
     ext_modules=ext_modules,
     # extras_require={"test": "pytest"},
     # Currently, build_ext only provides an optional "highest supported C++