diff --git a/.github/workflows/maturin-ci.yml b/.github/workflows/maturin-ci.yml new file mode 100644 index 0000000..8268eaf --- /dev/null +++ b/.github/workflows/maturin-ci.yml @@ -0,0 +1,120 @@ +# This file is autogenerated by maturin v1.1.0 +# To update, run +# +# maturin generate-ci github +# +name: CI + +on: + push: + branches: + - main + - master + tags: + - '*' + pull_request: + workflow_dispatch: + +permissions: + contents: read + +jobs: + linux: + runs-on: ubuntu-latest + strategy: + matrix: + target: [x86_64, x86, aarch64, armv7, s390x, ppc64le] + steps: + - uses: actions/checkout@v3 + - uses: actions/setup-python@v4 + with: + python-version: '3.10' + - name: Build wheels + uses: PyO3/maturin-action@v1 + with: + target: ${{ matrix.target }} + args: --release --out dist --find-interpreter + sccache: 'true' + manylinux: auto + - name: Upload wheels + uses: actions/upload-artifact@v3 + with: + name: wheels + path: dist + + windows: + runs-on: windows-latest + strategy: + matrix: + target: [x64, x86] + steps: + - uses: actions/checkout@v3 + - uses: actions/setup-python@v4 + with: + python-version: '3.10' + architecture: ${{ matrix.target }} + - name: Build wheels + uses: PyO3/maturin-action@v1 + with: + target: ${{ matrix.target }} + args: --release --out dist --find-interpreter + sccache: 'true' + - name: Upload wheels + uses: actions/upload-artifact@v3 + with: + name: wheels + path: dist + + macos: + runs-on: macos-latest + strategy: + matrix: + target: [x86_64, aarch64] + steps: + - uses: actions/checkout@v3 + - uses: actions/setup-python@v4 + with: + python-version: '3.10' + - name: Build wheels + uses: PyO3/maturin-action@v1 + with: + target: ${{ matrix.target }} + args: --release --out dist --find-interpreter + sccache: 'true' + - name: Upload wheels + uses: actions/upload-artifact@v3 + with: + name: wheels + path: dist + + sdist: + runs-on: ubuntu-latest + steps: + - uses: actions/checkout@v3 + - name: Build sdist + uses: PyO3/maturin-action@v1 + with: + command: sdist + args: --out dist + - name: Upload sdist + uses: actions/upload-artifact@v3 + with: + name: wheels + path: dist + + release: + name: Release + runs-on: ubuntu-latest + if: "startsWith(github.ref, 'refs/tags/')" + needs: [linux, windows, macos, sdist] + steps: + - uses: actions/download-artifact@v3 + with: + name: wheels + - name: Publish to PyPI + uses: PyO3/maturin-action@v1 + env: + MATURIN_PYPI_TOKEN: ${{ secrets.PYPI_API_TOKEN }} + with: + command: upload + args: --skip-existing * diff --git a/Cargo.toml b/Cargo.toml index ef7608a..bbd1ad1 100644 --- a/Cargo.toml +++ b/Cargo.toml @@ -6,6 +6,10 @@ authors = [ "Guilherme Lázari ", "Pedro Tabacof ", ] +description = "Decision trees in Rust" +license = "MIT" +categories = ["machine-learning", "decision-trees", "python"] +keywords = ["machine-learning", "decision-trees", "python"] [lib] name = "rustrees" diff --git a/README.md b/README.md index b1240d6..6232127 100644 --- a/README.md +++ b/README.md @@ -98,5 +98,3 @@ And then import the library in the notebook: ```python import rustrees.decision_tree as rt_dt ``` - -``` diff --git a/pyproject.toml b/pyproject.toml index 943b9fe..9cc7b45 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,3 +1,11 @@ +[project] +name = "rustrees" +dependencies = [ + "numpy>=1.0", + "pandas>=2.0", + "scikit-learn>=1.0", + "pyarrow>=12.0", +] [tool.maturin] python-source = "python"