Spaces:
Sleeping
Sleeping
File size: 911 Bytes
ce7bf5b |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 |
import os
import tempfile
import pytest
import chroma
import chroma.utility.api as api
from chroma.models.graph_backbone import GraphBackbone
from chroma.utility.model import load_model
KEY_PATH = os.path.dirname(os.path.dirname(chroma.__file__))
KEY_PATH = os.path.join(KEY_PATH, "config.json")
@pytest.mark.skipif(not os.path.exists(KEY_PATH), reason="requires file.txt")
def test_api():
# Test Key Registration
with tempfile.TemporaryDirectory() as key_directory:
api.register_key("my_key", key_directory)
# Test Reading
api.read_key()
# Test Download
api.download_from_generate(
"https://chroma-weights.generatebiomedicines.com/", "chroma_backbone_v1.0.pt"
)
# Test Public Loading of BB Model (load a specific model using this requests pull)
model = load_model(
"named:public", GraphBackbone, device="cpu", strict_unexpected=False,
)
|