Skip to content

Commit

Permalink
let's get minimal
Browse files Browse the repository at this point in the history
  • Loading branch information
Harsha-Nori committed Jan 28, 2025
1 parent 3d49fae commit f47ccc0
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 4 deletions.
1 change: 1 addition & 0 deletions setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -63,6 +63,7 @@
"types-regex",
"types-requests",
"types-jsonschema",
"requests",
]
test_requires = [
"azure-identity",
Expand Down
5 changes: 1 addition & 4 deletions tests/conftest.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,20 +4,17 @@
import tempfile
import time
import uuid

import pytest
import requests

from guidance import models

# Ensure that asserts from tests/utils.py are rewritten by pytest to show helpful messages
pytest.register_assert_rewrite("tests.utils")

from ._llms_for_testing import AVAILABLE_MODELS
from .utils import get_model

SELECTED_MODEL_ENV_VARIABLE = "GUIDANCE_SELECTED_MODEL"

AVAILABLE_MODELS = {}

def pytest_addoption(parser):
default_model = os.getenv(SELECTED_MODEL_ENV_VARIABLE, "transformers_gpt2_cpu")
Expand Down

0 comments on commit f47ccc0

Please sign in to comment.