Merge pull request #1 from rskottap/master

Make tests work on any machine

Makefile

@@ -8,7 +8,7 @@ install: build
 	pip install dist/*.tar.gz
 
 develop:
-	pip install -e .
+	pip install -e .[develop]
 
 check:
 	pytest -v tests

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"

tests/test_types.py

@@ -1,13 +1,14 @@
 import os
+import gpts
 import pytest
 from typing import Annotated
 from guidance import block
 from guidance.models import LlamaCpp
 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():