From 55b34708500dd76d953ab258ec5f9a574aea52f5 Mon Sep 17 00:00:00 2001 From: Ishaan Mathur Date: Thu, 18 Sep 2025 16:19:58 +0000 Subject: [PATCH 1/2] pyright ignore flash_attn import because it is guarded --- esm/layers/attention.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/esm/layers/attention.py b/esm/layers/attention.py index 564ef90c..964d4282 100644 --- a/esm/layers/attention.py +++ b/esm/layers/attention.py @@ -8,7 +8,7 @@ from esm.layers.rotary import RotaryEmbedding, TritonRotaryEmbedding try: - from flash_attn import flash_attn_varlen_qkvpacked_func + from flash_attn import flash_attn_varlen_qkvpacked_func # type: ignore except (ImportError, RuntimeError): flash_attn_varlen_qkvpacked_func = None From e1477b826acdcab952d7115af6479a79fa624b5d Mon Sep 17 00:00:00 2001 From: Ishaan Mathur Date: Thu, 18 Sep 2025 16:26:54 +0000 Subject: [PATCH 2/2] fix oss test, bump pixi version --- .github/workflows/ci.yml | 8 ++++---- tests/oss_pytests/test_oss_client.py | 4 ++-- 2 files changed, 6 insertions(+), 6 deletions(-) diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml index 53e317ec..9f467996 100644 --- a/.github/workflows/ci.yml +++ b/.github/workflows/ci.yml @@ -25,9 +25,9 @@ jobs: uses: actions/checkout@v4 - name: Setup Environment - uses: prefix-dev/setup-pixi@v0.8.1 + uses: prefix-dev/setup-pixi@v0.9.0 with: - pixi-version: v0.47.0 + pixi-version: v0.54.0 cache: false cache-write: ${{ github.event_name == 'push' && github.ref_name == 'main' }} @@ -43,9 +43,9 @@ jobs: uses: actions/checkout@v4 - name: Setup Environment - uses: prefix-dev/setup-pixi@v0.8.1 + uses: prefix-dev/setup-pixi@v0.9.0 with: - pixi-version: v0.47.0 + pixi-version: v0.54.0 cache: false cache-write: ${{ github.event_name == 'push' && github.ref_name == 'main' }} diff --git a/tests/oss_pytests/test_oss_client.py b/tests/oss_pytests/test_oss_client.py index 6fa4a30d..9dd5b188 100644 --- a/tests/oss_pytests/test_oss_client.py +++ b/tests/oss_pytests/test_oss_client.py @@ -53,7 +53,7 @@ def test_oss_esm3_client(): def test_oss_esmc_client(): assert URL is not None - sequence = "MALWMRLLPLLALLALAVUUPDPAAA" + sequence = "MALWMRLLPLLALLALAVPDPAAA" model = "esmc-300m-2024-12" esmc_client = client(model=model, url=URL, token=API_TOKEN) @@ -75,7 +75,7 @@ def test_oss_esmc_client(): def test_oss_sequence_structure_forge_inference_client(): assert URL is not None - sequence = "MALWMRLLPLLALLALAVUUPDPAAA" + sequence = "MALWMRLLPLLALLALAVPDPAAA" model = "esm3-small-2024-03" client = SequenceStructureForgeInferenceClient( model=model, url=URL, token=API_TOKEN