Reproducible environments (#95)

This PR aims to minimize energy needed to get `b3d` running on different
machines. See [B3D
reproducibility](https://linear.app/chi-fro/project/b3d-reproducibility-2fab178a363f/overview)
for context.

Author: eightysteele

.github/workflows/pdoc.yml

@@ -0,0 +1,64 @@
+name: docs
+
+# https://linear.app/chi-fro/issue/GEN-443/parallelhoss-config-check
+# on:
+#   pull_request:
+#   push:
+#     branches:
+#       - main
+#   workflow_dispatch:
+
+permissions:
+  contents: write
+  pages: write
+  id-token: write
+
+jobs:
+  build-deploy:
+    strategy:
+      matrix:
+        runner: [ParallelHoss]
+    runs-on: ${{ matrix.runner }}
+    steps:
+      - uses: actions/checkout@v4
+
+      - id: auth
+        uses: google-github-actions/auth@v2
+        with:
+          credentials_json: "${{ secrets.ARTIFACT_REGISTRY_KEY }}"
+
+      - name: Set up Cloud SDK
+        uses: google-github-actions/setup-gcloud@v2
+
+      - name: pixi run docs
+        run: |
+          # install pixi
+          curl -fsSL https://pixi.sh/install.sh | bash
+          export PATH=/home/runner/.pixi/bin:$PATH
+
+          # install pipx and inject gcp backend
+          pixi global install pipx
+          pipx install keyring
+          pipx inject keyring \
+          keyrings.google-artifactregistry-auth \
+            --index-url \
+            https://pypi.org/simple
+
+          # build docs
+          pixi run docs
+
+      - name: Prepend timestamp to index.html
+        run: |
+          echo "Published at: $(date)<br/>" | cat - ./docs/static/index.html > temp && mv temp ./docs/static/index.html
+
+      - name: Create website artifact
+        uses: actions/upload-artifact@v4
+        with:
+          name: website
+          path: ./docs/static
+
+  publish:
+    needs: build-deploy
+    uses: probcomp/gen-website-private/.github/workflows/publish_private_website.yml@main
+    with:
+      artifact: website

.github/workflows/pixi.yml

@@ -0,0 +1,54 @@
+name: pixi
+
+on:
+  pull_request:
+  push:
+    branches:
+      - main
+
+jobs:
+  pixi-update:
+    runs-on: ubuntu-latest
+    steps:
+      - uses: actions/checkout@v4
+
+      - name: Authenticate gcloud
+        id: auth
+        uses: google-github-actions/auth@v2
+        with:
+          credentials_json: "${{ secrets.ARTIFACT_REGISTRY_KEY }}"
+
+      - name: Setup gcloud
+        uses: google-github-actions/setup-gcloud@v2
+
+      - name: Run pixi update
+        run: |
+          # install pixi
+          curl -fsSL https://pixi.sh/install.sh | bash
+          export PATH=/home/runner/.pixi/bin:$PATH
+
+          # install pipx and inject gcp backend
+          pixi global install pipx
+          pipx install keyring
+          pipx inject keyring \
+            keyrings.google-artifactregistry-auth \
+            --index-url \
+            https://pypi.org/simple
+
+          # ensure all deps across all environments and platforms can solve
+          pixi install --locked \
+            -v \
+            --pypi-keyring-provider \
+            subprocess
+
+          pixi install --locked \
+            --environment cpu \
+            -v \
+            --pypi-keyring-provider \
+            subprocess
+
+          pixi install --locked \
+            --environment gpu \
+            -v \
+            --pypi-keyring-provider \
+            subprocess

.github/workflows/shfmt.yml

@@ -0,0 +1,20 @@
+name: shfmt
+
+on:
+  pull_request:
+  push:
+    branches:
+      - main
+
+jobs:
+  shfmt:
+    runs-on: ubuntu-latest
+    steps:
+    - uses: actions/checkout@v4
+
+    - name: Run the sh-checker
+      uses: luizm/action-sh-checker@master
+      env:
+        SHFMT_OPTS: --write --simplify --indent 2
+      with:
+        sh_checker_shellcheck_disable: true

.gitignore

@@ -3,6 +3,7 @@
 *.png
 assets/shared_data_bucket/*
 assets/test_results/*
+!assets/test_results/README.md
 assets/bop/*
 assets/nuscenes/*
 assets/ycbineoat/*
@@ -11,10 +12,13 @@ assets/ycbineoat/*
 **/**.npz
 **/**.log
 **/**.json
+!torch.json
 **/**.npy
 **/**.txt
 !src/b3d/renderer/**/**.txt
 assets/kitti/*
 **/**.pth
-**/**.obj
-**/**.r3d
+.pixi/envs/*
+.#*
+__pycache__/
+*.py[cod]

.pixi/config.toml

@@ -0,0 +1,4 @@
+[pypi-config]
+index-url = "https://pypi.org/simple"
+extra-index-urls = ["https://[email protected]/probcomp-caliban/probcomp/simple"]
+keyring-provider = "subprocess"

.pre-commit-config.yaml

@@ -1,5 +1,5 @@
 ci:
-    skip: [ruff]
+    skip: [ruff, pixi-update-default, pixi-update-cpu, pixi-update-gpu]
 
 repos:
   - repo: https://github.com/pre-commit/pre-commit-hooks
@@ -19,3 +19,45 @@ repos:
 
       - id: ruff-format
         types_or: [ python, pyi ]
+
+  - repo: https://github.com/pecigonzalo/pre-commit-shfmt
+    rev: v2.2.0
+    hooks:
+      - id: shell-fmt-go
+        args:
+          - --write
+          - --simplify
+          - --indent
+          - "2"
+
+  - repo: local
+    hooks:
+      - id: pixi-update-default
+        name: pixi update default
+        entry: bash -c 'cd $(git rev-parse --show-toplevel) && pixi update && git add pixi.lock'
+        language: system
+        files: ^pyproject\.toml$
+        types: [text]
+
+      - id: pixi-update-cpu
+        name: pixi update cpu
+        entry: bash -c 'cd $(git rev-parse --show-toplevel) && pixi update -e cpu && git add pixi.lock'
+        language: system
+        files: ^pyproject\.toml$
+        types: [text]
+
+      - id: pixi-update-gpu
+        name: pixi update gpu
+        entry: bash -c 'cd $(git rev-parse --show-toplevel) && pixi update -e gpu && git add pixi.lock'
+        language: system
+        files: ^pyproject\.toml$
+        types: [text]
+
+  - repo: local
+    hooks:
+      - id: gcp-sh-test
+        name: gcp.sh tests
+        entry: bash -c 'cd $(git rev-parse --show-toplevel) && cd scripts && ./test-gcp.sh'
+        language: system
+        files: .*gcp.*\.sh$
+        types: [executable]

README.md

@@ -0,0 +1,7 @@
+<!doctype html>
+<html>
+<head>
+    <meta charset="utf-8">
+    <meta http-equiv="refresh" content="0; url=./b3d.html"/>
+</head>
+</html>