"""Install the bundled wheel, then load the public A3 encoder."""
from grasp import GRASPEncoder

encoder=GRASPEncoder.from_pretrained("caithmac/GRASP-A3")
vectors=encoder.encode(["CCO","N[C@@H](C)C(=O)O"])
assert vectors.shape==(2,768)
print(vectors.shape)