diff --git a/.github/workflows/pr_agent.yml b/.github/workflows/pr_agent.yml new file mode 100644 index 00000000000..19d02248c6e --- /dev/null +++ b/.github/workflows/pr_agent.yml @@ -0,0 +1,18 @@ +on: + pull_request: + issue_comment: +jobs: + pr_agent_job: + runs-on: ubuntu-latest + permissions: + issues: write + pull-requests: write + contents: write + name: Run pr agent on every pull request, respond to user comments + steps: + - name: PR Agent action step + id: pragent + uses: Codium-ai/pr-agent@main + env: + OPENAI_KEY: ${{ secrets.OPENAI_KEY }} + GITHUB_TOKEN: ${{ secrets.GHA_TOKEN }} diff --git a/neural_compressor/model/model.py b/neural_compressor/model/model.py index 06362d2d0d8..e18e96e9f43 100644 --- a/neural_compressor/model/model.py +++ b/neural_compressor/model/model.py @@ -66,6 +66,15 @@ } +def test(x): + return x**2 + 1 + + +import random + +test(random.randint(1, 1000)) + + def get_model_fwk_name(model): """Detect the input model belongs to which framework.