Skip to content

GitLab

"...text-generation-inference.git" did not exist on "befd9f6735ed8d7f5d8e9110b1f921e16d856a8b"
Unverified Commit 47c2b947 authored by Michael Yang's avatar Michael Yang Committed by GitHub
Browse files

Merge pull request #6546 from ollama/mxyng/fix-test 

fix(test): do not clobber models directory
parents 5eb77bf9 e4d0a9c3
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
server/model_test.go
View file @ 47c2b947
...@@ -139,6 +139,7 @@ The temperature in San Francisco, CA is 70°F and in Toronto, Canada is 20°C.`, ...@@ -139,6 +139,7 @@ The temperature in San Francisco, CA is 70°F and in Toronto, Canada is 20°C.`,
func TestParseFromFileFromLayer(t *testing.T) { func TestParseFromFileFromLayer(t *testing.T) {
tempModels := t.TempDir() tempModels := t.TempDir()
t.Setenv("OLLAMA_MODELS", tempModels)
file, err := os.CreateTemp(tempModels, "") file, err := os.CreateTemp(tempModels, "")
if err != nil { if err != nil {
...@@ -189,6 +190,7 @@ func TestParseFromFileFromLayer(t *testing.T) { ...@@ -189,6 +190,7 @@ func TestParseFromFileFromLayer(t *testing.T) {
func TestParseLayerFromCopy(t *testing.T) { func TestParseLayerFromCopy(t *testing.T) {
tempModels := t.TempDir() tempModels := t.TempDir()
t.Setenv("OLLAMA_MODELS", tempModels)
file2, err := os.CreateTemp(tempModels, "") file2, err := os.CreateTemp(tempModels, "")
if err != nil { if err != nil {
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment