Merge pull request #80 from graphcore-research/pypi

Switch to pyproject.toml

Author: thecharlieblake

.dockerignore

@@ -1,2 +1,2 @@
 *
-!requirements*.txt
+!requirements-dev.txt

.github/workflows/ci.yaml

@@ -18,12 +18,14 @@ jobs:
       - name: Checkout code
         uses: actions/checkout@v3
 
-      - name: Build Docker Image
-        run: |
-          docker build -t unit-scaling-dev:latest . 
+      - name: Build Docker image
+        run: docker build -t unit-scaling-dev:latest .
+
+      - name: Local unit_scaling install
+        run: docker run -v $(pwd):/home/developer/unit-scaling unit-scaling-dev:latest pip install --user -e .
 
       - name: Run CI
-        run: docker run --rm -v $(pwd):/home/developer/unit-scaling unit-scaling-dev:latest ./dev ci
+        run: docker run -v $(pwd):/home/developer/unit-scaling unit-scaling-dev:latest ./dev ci
 
       - name: Publish documentation
         if: ${{github.ref == 'refs/heads/main'}}

.gitignore

@@ -7,6 +7,7 @@ __pycache__
 .venv
 .venvs
 .vscode
+unit_scaling/_version.py
 
 /build
 /dist

Dockerfile

@@ -31,7 +31,7 @@ ENV PATH="$PATH:/home/$USERNAME/.local/bin" \
 
 # Install Python dependencies
 COPY requirements-dev.txt .
-RUN pip install -r requirements-dev.txt
+RUN pip install --user -r requirements-dev.txt
 
 # Creates basic .zshrc
 RUN sudo cp /etc/zsh/newuser.zshrc.recommended /home/$USERNAME/.zshrc

MANIFEST.in

@@ -2,7 +2,7 @@ Copyright (c) 2023 Graphcore Ltd. Licensed under the Apache 2.0 License.
 
 The included code is released under an Apache 2.0 license, (see [LICENSE](LICENSE)).
 
-Our dependencies are (see [requirements.txt](requirements.txt)):
+Our dependencies are (see [pyproject.toml](pyproject.toml)):
 
 | Component | About | License |
 | --- | --- | --- |

NOTICE.md

@@ -1,5 +1,10 @@
 # Unit-Scaled Maximal Update Parameterization (u-μP)
 
+[![tests](https://github.com/graphcore-research/unit-scaling/actions/workflows/ci.yaml/badge.svg)](https://github.com/graphcore-research/unit-scaling/actions/workflows/ci-public.yaml)
+![PyPI version](https://img.shields.io/pypi/v/unit-scaling)
+[![license](https://img.shields.io/badge/License-Apache%202.0-blue.svg)](https://github.com/graphcore-research/unit-scaling/blob/main/LICENSE)
+[![GitHub Repo stars](https://img.shields.io/github/stars/graphcore-research/unit-scaling)](https://github.com/graphcore-research/unit-scaling/stargazers)
+
 A library for unit scaling in PyTorch, based on the paper [u-μP: The Unit-Scaled Maximal Update Parametrization](https://arxiv.org/abs/2407.17465) and previous work [Unit Scaling: Out-of-the-Box Low-Precision Training](https://arxiv.org/abs/2303.11257).
 
 Documentation can be found at
@@ -14,7 +19,12 @@ We're keen to help users with any problems they encounter.
 To install the `unit-scaling` library, run:
 
 ```sh
-pip install git+https://github.com/graphcore-research/unit-scaling.git
+pip install unit-scaling
+```
+or for a local editable install (i.e. one which uses the files in this repo), run:
+
+```sh
+pip install -e .
 ```
 
 ## Development

README.md

@@ -85,8 +85,8 @@ def format(check: bool) -> None:
 def copyright() -> None:
     """check for Graphcore copyright headers on relevant files"""
     command = (
-        f"find {' '.join(PYTHON_ROOTS)} -type f"
-        " -not -name *.pyc -not -name *.json -not -name .gitignore"
+        f"find {' '.join(PYTHON_ROOTS)} -type f -not -name *.pyc -not -name *.json"
+        " -not -name .gitignore -not -name *_version.py"
         " | xargs grep -L 'Copyright (c) 202. Graphcore Ltd[.] All rights reserved[.]'"
     )
     print(f"$ {command}", file=sys.stderr)

dev

@@ -18,7 +18,7 @@ To install :code:`unit-scaling`, run:
 
 .. code-block::
 
-    pip install git+https://github.com/graphcore-research/unit-scaling.git
+    pip install unit-scaling
 
 Getting Started
 ---------------

docs/index.rst

@@ -19,7 +19,7 @@ To install the :code:`unit-scaling` library, run:
 
 .. code-block::
 
-    pip install git+https://github.com/graphcore-research/unit-scaling.git
+    pip install unit-scaling
 
 For those who wish to develop on the :code:`unit-scaling` codebase, clone or fork our
 `GitHub repo <https://github.com/graphcore-research/unit-scaling.git>`_ and follow the