From 8997ff6042397f1ace41defa3b3e4e40b91c17ce Mon Sep 17 00:00:00 2001 From: Ettore Di Giacinto Date: Sat, 21 Mar 2026 01:04:28 +0000 Subject: [PATCH] Fix tests Signed-off-by: Ettore Di Giacinto --- core/gallery/importers/local_test.go | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/core/gallery/importers/local_test.go b/core/gallery/importers/local_test.go index b43433406..0de679462 100644 --- a/core/gallery/importers/local_test.go +++ b/core/gallery/importers/local_test.go @@ -68,7 +68,7 @@ var _ = Describe("ImportLocalPath", func() { Expect(err).ToNot(HaveOccurred()) Expect(cfg.Backend).To(Equal("transformers")) Expect(cfg.Model).To(Equal("meta-llama/Llama-2-7b-hf")) - Expect(cfg.LLMConfig.LoraAdapter).To(Equal(modelDir)) + Expect(cfg.LLMConfig.LoraAdapter).To(Equal("lora-model")) Expect(cfg.TemplateConfig.UseTokenizerTemplate).To(BeTrue()) }) @@ -100,7 +100,7 @@ var _ = Describe("ImportLocalPath", func() { cfg, err := importers.ImportLocalPath(modelDir, "merged") Expect(err).ToNot(HaveOccurred()) Expect(cfg.Backend).To(Equal("transformers")) - Expect(cfg.Model).To(Equal(modelDir)) + Expect(cfg.Model).To(Equal("merged")) Expect(cfg.TemplateConfig.UseTokenizerTemplate).To(BeTrue()) }) @@ -113,7 +113,7 @@ var _ = Describe("ImportLocalPath", func() { cfg, err := importers.ImportLocalPath(modelDir, "merged-pt") Expect(err).ToNot(HaveOccurred()) Expect(cfg.Backend).To(Equal("transformers")) - Expect(cfg.Model).To(Equal(modelDir)) + Expect(cfg.Model).To(Equal("merged-pt")) }) })