Skip to content

Commit bba9e61

Browse files
author
Fausto Milletari
committed
Revert "Fix test_oss_esmc_client to use esmc_client() factory"
This reverts commit 2f521b7.
1 parent 2f521b7 commit bba9e61

1 file changed

Lines changed: 5 additions & 5 deletions

File tree

tests/oss_pytests/test_oss_client.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@
33
import pytest
44
import torch
55

6-
from esm.sdk import client, esmc_client # pyright: ignore
6+
from esm.sdk import client # pyright: ignore
77
from esm.sdk.api import ( # pyright: ignore
88
ESMProtein,
99
ESMProteinTensor,
@@ -58,19 +58,19 @@ def test_oss_esmc_client():
5858

5959
sequence = "MALWMRLLPLLALLALAVPDPAAA"
6060
model = "esmc-300m-2024-12"
61-
esmc = esmc_client(model=model, url=URL, token=API_TOKEN)
61+
esmc_client = client(model=model, url=URL, token=API_TOKEN)
6262

6363
protein = ESMProtein(sequence)
64-
encoded_protein = esmc.encode(input=protein)
64+
encoded_protein = esmc_client.encode(input=protein)
6565
assert isinstance(encoded_protein, ESMProteinTensor)
6666

67-
decoded_protein = esmc.decode(input=encoded_protein)
67+
decoded_protein = esmc_client.decode(input=encoded_protein)
6868
assert isinstance(decoded_protein, ESMProtein)
6969

7070
logits_config = LogitsConfig(
7171
sequence=True, return_embeddings=True, return_hidden_states=True
7272
)
73-
result = esmc.logits(input=encoded_protein, config=logits_config)
73+
result = esmc_client.logits(input=encoded_protein, config=logits_config)
7474
assert isinstance(result, LogitsOutput)
7575
assert result.logits is not None
7676
assert isinstance(result.logits.sequence, torch.Tensor)

0 commit comments

Comments
 (0)