Python 3.10 support (#13)

* step down to 3.10 * matrix test 3.10 * update changelog * add python version badge and some other badges to README
vgel · Mar 5, 2024 · 4b840f2 · 4b840f2
1 parent ffcfb3c
commit 4b840f2
Show file tree

Hide file tree

Showing 6 changed files with 39 additions and 6 deletions.
diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
@@ -19,7 +19,7 @@ jobs:
     runs-on: ubuntu-latest
     strategy:
       matrix:
-        python-version: ["3.11"]
+        python-version: ["3.10", "3.11"]
 
     steps:
     - uses: actions/checkout@v2

diff --git a/CHANGELOG b/CHANGELOG
@@ -3,6 +3,7 @@
 ## Unreleased
 
 * Add GPT-2 support. (#12)
+* Officially support Python 3.10 and 3.11. (#13)
 
 ## 0.2.0 - 2024-03-03
 

diff --git a/README.md b/README.md
@@ -1,6 +1,9 @@
 # repeng
 
-[![PyPI version](https://badge.fury.io/py/repeng.svg)](https://badge.fury.io/py/repeng)
+![GitHub Actions Workflow Status](https://img.shields.io/github/actions/workflow/status/vgel/repeng/ci.yml?label=ci)
+![PyPI - Version](https://img.shields.io/pypi/v/repeng)
+![PyPI - Python Version](https://img.shields.io/pypi/pyversions/repeng)
+![GitHub License](https://img.shields.io/github/license/vgel/repeng)
 
 A Python library for generating control vectors with representation engineering.
 Train a vector in less than sixty seconds!

diff --git a/notebooks/vector_ops.ipynb b/notebooks/vector_ops.ipynb
@@ -549,7 +549,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.11.7"
+   "version": "3.10.12"
   }
  },
  "nbformat": 4,

diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -6,7 +6,7 @@ authors = ["Theia Vogel <theia@vgel.me>"]
 readme = "README.md"
 
 [tool.poetry.dependencies]
-python = "^3.11"
+python = ">=3.10,<3.12" # `accelerate` doesn't support 3.12 yet
 numpy = "^1.26.3"
 scikit-learn = "^1.4.0"
 torch = "^2.1.2"