diff --git a/Makefile b/Makefile index f81dc6e..cb3ce2d 100644 --- a/Makefile +++ b/Makefile @@ -8,7 +8,7 @@ install: build pip install dist/*.tar.gz develop: - pip install -e . + pip install -e .[develop] check: pytest -v tests diff --git a/pyproject.toml b/pyproject.toml index b99593c..23534fe 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -7,11 +7,17 @@ authors = [ description = "Remove the yammering from LLM outputs." dependencies = [ "pyparsing", - "pytest", "pydantic", "guidance", ] +[project.optional-dependencies] +develop = [ + "pytest", + "build", + "gpts", +] + [project.urls] source = "https://github.com/hudson-ai/minml/" download = "https://pypi.org/project/minml/#files" diff --git a/tests/test_types.py b/tests/test_types.py index 4360f32..3094d46 100644 --- a/tests/test_types.py +++ b/tests/test_types.py @@ -1,4 +1,5 @@ import os +import gpts import pytest from typing import Annotated from guidance import block @@ -6,8 +7,8 @@ from pydantic import BaseModel, TypeAdapter, StringConstraints from minml import types -MODEL_FILE = os.path.expanduser("~/pkg/mistral/mistral-7b-instruct.gguf") +MODEL_FILE = gpts.Mistral().path() @pytest.fixture(scope="session") def model():