Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
55 changes: 55 additions & 0 deletions .github/workflows/pytest.yml
Original file line number Diff line number Diff line change
@@ -0,0 +1,55 @@
name: Pytest

on:
push:
branches: ["*"]
pull_request:
branches: ["*"]

jobs:
test:
runs-on: ubuntu-latest
strategy:
matrix:
python-version: ["3.11"]
name: run pytest (Python ${{ matrix.python-version }})
steps:
- name: Checkout project
uses: actions/checkout@v4
with:
fetch-depth: 0

- name: Install uv
uses: astral-sh/setup-uv@v4
with:
version: "latest"

- name: Set up Python ${{ matrix.python-version }}
uses: actions/setup-python@v5
with:
python-version: ${{ matrix.python-version }}

- name: Create virtual environment
run: |
uv venv

- name: Set up elan
run: |
curl https://raw.githubusercontent.com/leanprover/elan/master/elan-init.sh -sSf | sh -s -- -y
echo "$HOME/.elan/bin" >> $GITHUB_PATH

- name: Install Pantograph
run: |
uv pip install git+https://github.com/stanford-centaur/PyPantograph

- name: Install project with dev dependencies
run: |
uv pip install -e ".[dev]"

- name: Run pytest
env:
GITHUB_ACCESS_TOKEN: ${{ secrets.GITHUB_ACCESS_TOKEN }}
run: |
source .venv/bin/activate
pytest -vvv

2 changes: 1 addition & 1 deletion lean_dojo_v2/external_api/lean-toolchain
Original file line number Diff line number Diff line change
@@ -1 +1 @@
leanprover/lean4:v4.21.0-rc3
leanprover/lean4:v4.24.0
5 changes: 0 additions & 5 deletions lean_dojo_v2/external_api/python/external_models/__init__.py
Original file line number Diff line number Diff line change
@@ -1,6 +1 @@
# from .claude_runner import ClaudeRunner
# from .gemini_runner import GeminiRunner
from .hf_runner import HFTacticGenerator

# from .oai_runner import OpenAIRunner
# from .vllm_runner import VLLMTacticGenerator
45 changes: 0 additions & 45 deletions lean_dojo_v2/external_api/python/external_models/claude_runner.py

This file was deleted.

73 changes: 0 additions & 73 deletions lean_dojo_v2/external_api/python/external_models/gemini_runner.py

This file was deleted.

83 changes: 0 additions & 83 deletions lean_dojo_v2/external_api/python/external_models/oai_runner.py

This file was deleted.

84 changes: 0 additions & 84 deletions lean_dojo_v2/external_api/python/external_models/vllm_runner.py

This file was deleted.

17 changes: 1 addition & 16 deletions lean_dojo_v2/external_api/python/server.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,29 +4,14 @@
import torch
from external_models import *
from fastapi import FastAPI, HTTPException
from loguru import logger
from models import *
from pydantic import BaseModel

app = FastAPI()

models = {
"kimina": VLLMTacticGenerator(
model="AI-MO/Kimina-Prover-Preview-Distill-7B",
tensor_parallel_size=1,
temperature=0.6,
max_tokens=1024,
top_p=0.9,
length_penalty=0,
n=4,
do_sample=True,
output_scores=True,
output_logits=False,
return_dict_in_generate=True,
device="auto",
),
"deepseek": HFTacticGenerator(
model="deepseek-ai/DeepSeek-Prover-V2-7B",
model="deepseek-ai/DeepSeek-Prover-V2-671B:novita",
temperature=0.6,
max_new_tokens=256,
top_p=0.9,
Expand Down
Loading