Skip to content

Commit

Permalink
Use a script to automatically set the version in Cargo.toml
Browse files Browse the repository at this point in the history
  • Loading branch information
rlouf committed Dec 9, 2024
1 parent c74f157 commit 9fec454
Show file tree
Hide file tree
Showing 4 changed files with 49 additions and 2 deletions.
40 changes: 40 additions & 0 deletions .github/scripts/cargo_version_bumper.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,40 @@
# From https://github.com/Intreecom/scyllapy/blob/05fdab32dd7468c26533de5fdfe9627fa3e38445/scripts/version_bumper.py

import re
import argparse
from pathlib import Path


def parse_args() -> argparse.Namespace:
"""Parse command line arguments."""
parser = argparse.ArgumentParser()
parser.add_argument(
"--target",
"-t",
dest="target",
type=Path,
default="Cargo.toml",
)
parser.add_argument("version", type=str)
return parser.parse_args()


def main() -> None:
"""Main function."""
args = parse_args()
with args.target.open("r") as f:
contents = f.read()

contents = re.sub(
r"version\s*=\s*\"(.*)\"",
f'version = "{args.version}"',
contents,
count=1,
)

with args.target.open("w") as f:
f.write(contents)


if __name__ == "__main__":
main()
5 changes: 4 additions & 1 deletion .github/workflows/release_pypi.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -23,6 +23,10 @@ jobs:
with:
python-version: '3.10'

- name: Bump Cargo version
run: |
python ./.github/scripts/cargo_version_bumper.py --target Cargo.toml "${{ github.ref_name }}"
- name: Set up QEMU
if: runner.os == 'Linux'
uses: docker/setup-qemu-action@v3
Expand Down Expand Up @@ -71,7 +75,6 @@ jobs:
CIBW_TEST_COMMAND: python -c "import outlines_core; print(outlines_core.__version__)"
CMAKE_PREFIX_PATH: ./dist


- uses: actions/upload-artifact@v3
with:
path: ./wheelhouse/*.whl
Expand Down
4 changes: 4 additions & 0 deletions .github/workflows/release_rust.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -15,6 +15,10 @@ jobs:
- name: Checkout
uses: actions/checkout@v2

- name: Bump Cargo version
run: |
python ./.github/scripts/cargo_version_bumper.py --target Cargo.toml "${{ github.ref_name }}"
- name: Install Rust
uses: actions-rs/toolchain@v1
with:
Expand Down
2 changes: 1 addition & 1 deletion Cargo.toml
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
[package]
name = "outlines-core"
version = "0.1.21"
version = "0.0.0"
edition = "2021"
description = "Structured Generation"
license = "Apache-2.0"
Expand Down

0 comments on commit 9fec454

Please sign in to comment.