Skip to content

Commit 08944ed

Browse files
AverylampAverylamp
committed
Merge branch 'release/0.3.0' into main
2 parents 6dd0a0f + c72367c commit 08944ed

File tree

262 files changed

+3345
-5893
lines changed

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

262 files changed

+3345
-5893
lines changed

‎.github/ISSUE_TEMPLATE/---bug-report.md‎

Lines changed: 32 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,32 @@
1+
---
2+
name: "\U0001F41B Bug report"
3+
about: Submit a bug report to improve our library!
4+
title: ''
5+
labels: bug
6+
assignees: ''
7+
8+
---
9+
10+
** Environment **
11+
<!-- Provide the below as necessary -->
12+
13+
- OS: [Ubuntu 20.04]
14+
- Hardware (GPU, or instance type): [V100]
15+
16+
<!-- if your bug is CUDA related, please also provide your CUDA version -->
17+
18+
** To reproduce
19+
20+
Steps to reproduce the behavior:
21+
22+
1.
23+
2.
24+
3.
25+
26+
## Expected behavior
27+
28+
<!-- A clear and concise description of what you would expect to happen. -->
29+
30+
## Additional context
31+
32+
<!-- Please provide an additional context. -->

‎.github/ISSUE_TEMPLATE/---feature-request.md‎

Lines changed: 23 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,23 @@
1+
---
2+
name: "\U0001F680 Feature request"
3+
about: Suggest an idea for this project
4+
title: ''
5+
labels: enhancement
6+
assignees: ''
7+
8+
---
9+
10+
## 🚀 Feature Request
11+
<!-- A clear and concise description of the feature proposal -->
12+
13+
## Motivation
14+
15+
<!-- Please outline the motivation for the proposal. Is your feature request related to a problem? e.g., I'm always frustrated when [...]. If this is related to another GitHub issue, please link here too -->
16+
17+
## [Optional] Implementation
18+
19+
<!-- Optionally, sketch out an implementation or interface needed. -->
20+
21+
## Additional context
22+
23+
<!-- Add any other context or screenshots about the feature request here. -->

‎.github/ISSUE_TEMPLATE/---new-method.md‎

Lines changed: 23 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,23 @@
1+
---
2+
name: "\U0001F525 New Method"
3+
about: Request or submit an implementation of a new efficiency method
4+
title: ''
5+
labels: method
6+
assignees: ''
7+
8+
---
9+
10+
## New Method
11+
<!-- A tl;dr of the requested method, linking to papers and code as appropriate -->
12+
13+
## Motivation
14+
15+
<!-- Why is this method important? Please provide any expected gains in quality or accuracy from the literature or your own experiments. -->
16+
17+
## Attribution
18+
19+
<!-- Who are the authors that we should credit and/or contact for this method? -->
20+
21+
## [Optional] Implementation
22+
23+
<!-- Optionally, sketch out the desired implementation or interface for the method-->

‎.github/actions/isort/action.yaml‎

Lines changed: 27 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,27 @@
1+
name: ISort
2+
description: Sort imports
3+
inputs:
4+
working_directory:
5+
required: true
6+
description: Root of the repo. Should be set to GITHUB_WORKSPACE
7+
args:
8+
description: ISort args
9+
default: ". -c -v"
10+
required: false
11+
runs:
12+
using: composite
13+
steps:
14+
- name: Install ISort
15+
shell: bash
16+
working-directory: ${{ inputs.working_directory }}
17+
run: |
18+
set -ex
19+
pip install isort
20+
- name: Run ISort
21+
shell: bash
22+
working-directory: ${{ inputs.working_directory }}
23+
env:
24+
ISORT_ARGS: ${{ inputs.args }}
25+
run: |
26+
set -ex
27+
isort ${ISORT_ARGS}

‎.github/actions/license/action.yaml‎

Lines changed: 23 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,23 @@
1+
name: Check License Headers
2+
description: Ensure that all source files have the proper license headers
3+
inputs:
4+
working_directory:
5+
description: 'Working Directory'
6+
required: true
7+
runs:
8+
using: composite
9+
steps:
10+
- name: Download Addlicense
11+
shell: bash
12+
run: |
13+
set -ex
14+
wget -c https://github.com/google/addlicense/releases/download/v1.0.0/addlicense_1.0.0_Linux_x86_64.tar.gz -O - | sudo tar -xz -C /usr/local/bin
15+
- name: Run Addlicense
16+
shell: bash
17+
working-directory: ${{ inputs.working_directory }}
18+
env:
19+
WORKING_DIRECTORY: ${{ inputs.working_directory }}
20+
run: |
21+
set -ex
22+
# Recursively check that the license exists on all files
23+
find . -type f -not -path '*/\.*' \( -iname \*.py -o -iname \*.pyi \) -print0 | xargs -0 -n1 addlicense -check -f ${WORKING_DIRECTORY}/LICENSE_HEADER

‎.github/actions/pyright/action.yaml‎

Lines changed: 22 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,22 @@
1+
name: Pyright
2+
description: Run pyright type checker
3+
inputs:
4+
working_directory:
5+
required: true
6+
description: Directory containing the Makefile
7+
runs:
8+
using: composite
9+
steps:
10+
- name: Install Pyright
11+
shell: bash
12+
working-directory: ${{ inputs.working_directory }}
13+
run: |
14+
set -ex
15+
sudo npm install -g [email protected]
16+
17+
- name: Run Pyright
18+
shell: bash
19+
working-directory: ${{ inputs.working_directory }}
20+
run: |
21+
set -ex
22+
pyright

‎.github/actions/yapf/action.yaml‎

Lines changed: 27 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,27 @@
1+
name: YAPF
2+
description: Format with YAPF
3+
inputs:
4+
working_directory:
5+
required: true
6+
description: Root of the repo. Should be set to GITHUB_WORKSPACE
7+
args:
8+
description: YAPF args
9+
default: "-drp ."
10+
required: false
11+
runs:
12+
using: composite
13+
steps:
14+
- name: Install YAPF
15+
shell: bash
16+
working-directory: ${{ inputs.working_directory }}
17+
run: |
18+
set -ex
19+
pip install yapf
20+
- name: Run YAPF
21+
shell: bash
22+
working-directory: ${{ inputs.working_directory }}
23+
env:
24+
YAPF_ARGS: ${{ inputs.args }}
25+
run: |
26+
set -ex
27+
yapf ${YAPF_ARGS}

‎.github/workflows/formatting.yaml‎

Lines changed: 41 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,41 @@
1+
name: Formatting
2+
on:
3+
push: {}
4+
pull_request: {}
5+
workflow_dispatch: {}
6+
defaults:
7+
run:
8+
working-directory: .
9+
jobs:
10+
formatting:
11+
runs-on: ubuntu-latest
12+
timeout-minutes: 10
13+
steps:
14+
- uses: actions/checkout@v2
15+
- uses: actions/setup-python@v2
16+
with:
17+
python-version: 3.8
18+
- name: Cache pip
19+
uses: actions/cache@v2
20+
with:
21+
# This path is specific to Ubuntu
22+
path: ~/.cache/pip
23+
# Look to see if there is a cache hit for the corresponding requirements file
24+
key: ${{ runner.os }}-pip-${{ hashFiles('setup.py') }}
25+
restore-keys: |
26+
${{ runner.os }}-pip-
27+
${{ runner.os }}-
28+
- name: Setup
29+
run: |
30+
set -ex
31+
python -m pip install --upgrade pip wheel
32+
python -m pip install -e .[all]
33+
- uses: ./.github/actions/yapf
34+
with:
35+
working_directory: ${{ github.workspace }}
36+
- uses: ./.github/actions/isort
37+
with:
38+
working_directory: "${{ github.workspace }}"
39+
- uses: ./.github/actions/license
40+
with:
41+
working_directory: "${{ github.workspace }}"

‎.github/workflows/pr-gate-cpu.yaml‎

Lines changed: 37 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,37 @@
1+
name: PR Gate (CPU)
2+
on:
3+
push: {}
4+
pull_request: {}
5+
workflow_dispatch: {}
6+
jobs:
7+
runner-cpu:
8+
timeout-minutes: 30
9+
runs-on: ubuntu-latest
10+
steps:
11+
- uses: actions/checkout@v2
12+
- uses: actions/setup-python@v2
13+
with:
14+
python-version: 3.8
15+
- name: Cache pip
16+
uses: actions/cache@v2
17+
with:
18+
# This path is specific to Ubuntu
19+
path: ~/.cache/pip
20+
# Look to see if there is a cache hit for the corresponding requirements file
21+
key: ${{ runner.os }}-pip-${{ hashFiles('setup.py') }}
22+
restore-keys: |
23+
${{ runner.os }}-pip-
24+
${{ runner.os }}-
25+
- name: Setup
26+
run: |
27+
set -exuo pipefail
28+
python -m pip install --upgrade pip wheel
29+
30+
python -m pip install -e .[all]
31+
- name: Disable WandB
32+
run: wandb disabled
33+
- name: Run Tests
34+
id: tests
35+
run: |
36+
set -exuo pipefail
37+
./scripts/test.sh --test_duration all

‎.github/workflows/pyright.yaml‎

Lines changed: 35 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,35 @@
1+
name: Pyright
2+
on:
3+
push: {}
4+
pull_request: {}
5+
workflow_dispatch: {}
6+
defaults:
7+
run:
8+
working-directory: .
9+
jobs:
10+
pyright:
11+
runs-on: ubuntu-latest
12+
timeout-minutes: 10
13+
steps:
14+
- uses: actions/checkout@v2
15+
- uses: actions/setup-python@v2
16+
with:
17+
python-version: 3.8
18+
- name: Cache pip
19+
uses: actions/cache@v2
20+
with:
21+
# This path is specific to Ubuntu
22+
path: ~/.cache/pip
23+
# Look to see if there is a cache hit for the corresponding requirements file
24+
key: ${{ runner.os }}-pip-${{ hashFiles('setup.py') }}
25+
restore-keys: |
26+
${{ runner.os }}-pip-
27+
${{ runner.os }}-
28+
- name: Setup
29+
run: |
30+
set -ex
31+
python -m pip install --upgrade pip wheel
32+
python -m pip install -e .[all]
33+
- uses: ./.github/actions/pyright
34+
with:
35+
working_directory: ${{ github.workspace }}

0 commit comments

Comments
 (0)