fix nimparser unittests

Signed-off-by: Varun Ramachandra Sekar <vsekar@nvidia.com>

Author: varunrsekar

internal/nimparser/testdata/manifest_vllm.yaml

@@ -14,7 +14,7 @@ See the License for the specific language governing permissions and
 limitations under the License.
 */
 
-package nimparser_test
+package v1_test
 
 import (
 	"testing"
@@ -23,7 +23,7 @@ import (
 	. "github.com/onsi/gomega"
 )
 
-func TestRender(t *testing.T) {
+func TestNIMParser(t *testing.T) {
 	RegisterFailHandler(Fail)
 	RunSpecs(t, "Model Manifest Parser Test Suite")
 }

…ternal/nimparser/nimparser_suite_test.go …nal/nimparser/v1/nimparser_suite_test.gointernal/nimparser/nimparser_suite_test.go renamed to internal/nimparser/v1/nimparser_suite_test.go internal/nimparser/nimparser_suite_test.go renamed to internal/nimparser/v1/nimparser_suite_test.go

@@ -32,7 +32,7 @@ var _ = Describe("NIMParser", func() {
 	Context("ParseModelManifest", func() {
 		It("should parse a model profile for trtllm engine files correctly", func() {
 
-			filePath := filepath.Join("testdata", "manifest_trtllm.yaml")
+			filePath := filepath.Join("testdata", "manifest_trtllm_v1.yaml")
 			nimparser := NIMParser{}
 			config, err := nimparser.ParseModelManifest(filePath)
 			Expect(err).NotTo(HaveOccurred())
@@ -49,7 +49,7 @@ var _ = Describe("NIMParser", func() {
 		})
 		It("should parse a model profile for vllm engine files correctly", func() {
 
-			filePath := filepath.Join("testdata", "manifest_vllm.yaml")
+			filePath := filepath.Join("testdata", "manifest_vllm_v1.yaml")
 			nimparser := NIMParser{}
 			config, err := nimparser.ParseModelManifest(filePath)
 			Expect(err).NotTo(HaveOccurred())
@@ -83,7 +83,7 @@ var _ = Describe("NIMParser", func() {
 			Expect(profile.ContainerURL).To(Equal("nvcr.io/nim/meta/llama3-70b-instruct:1.0.0"))
 		})
 		It("should match model profiles with valid parameters", func() {
-			filePath := filepath.Join("testdata", "manifest_trtllm.yaml")
+			filePath := filepath.Join("testdata", "manifest_trtllm_v1.yaml")
 			nimparser := NIMParser{}
 			config, err := nimparser.ParseModelManifest(filePath)
 			Expect(err).NotTo(HaveOccurred())
@@ -112,7 +112,7 @@ var _ = Describe("NIMParser", func() {
 			Expect(matchedProfiles).To(HaveLen(1))
 		})
 		It("should not match model profiles with invalid parameters", func() {
-			filePath := filepath.Join("testdata", "manifest_trtllm.yaml")
+			filePath := filepath.Join("testdata", "manifest_trtllm_v1.yaml")
 			nimparser := NIMParser{}
 			config, err := nimparser.ParseModelManifest(filePath)
 			Expect(err).NotTo(HaveOccurred())
@@ -133,7 +133,7 @@ var _ = Describe("NIMParser", func() {
 			Expect(matchedProfiles).To(BeEmpty())
 		})
 		It("should match model profiles using automatically discovered GPUs", func() {
-			filePath := filepath.Join("testdata", "manifest_trtllm.yaml")
+			filePath := filepath.Join("testdata", "manifest_trtllm_v1.yaml")
 			nimparser := NIMParser{}
 			config, err := nimparser.ParseModelManifest(filePath)
 			Expect(err).NotTo(HaveOccurred())

internal/nimparser/v1/nimparser_test.go

@@ -11,3 +11,15 @@
     profile: throughput
     tp: '8'
   container_url: nvcr.io/nim/meta/llama3-70b-instruct:1.0.0
+  workspace: !workspace
+    components:
+    - dst: trtllm_engine
+      src:
+        files:
+          - !name 'LICENSE.txt'
+          - !name 'NOTICE.txt'
+          - !name 'checksums.blake3'
+          - !name 'config.json'
+          - !name 'metadata.json'
+          - !name 'rank0.engine'
+          - !name 'rank1.engine'

…al/nimparser/testdata/manifest_lora.yaml …nimparser/v1/testdata/manifest_lora.yamlinternal/nimparser/testdata/manifest_lora.yaml renamed to internal/nimparser/v1/testdata/manifest_lora.yaml internal/nimparser/testdata/manifest_lora.yaml renamed to internal/nimparser/v1/testdata/manifest_lora.yaml

@@ -0,0 +1,29 @@
+0f3de1afe11d355e01657424a267fbaad19bfea3143a9879307c49aed8299db0:
+  model: meta/llama3-70b-instruct
+  release: '1.0.0'
+  tags:
+    feat_lora: 'true'
+    feat_lora_max_rank: '32'
+    llm_engine: vllm
+    precision: fp16
+    tp: '8'
+  container_url: nvcr.io/nim/meta/llama3-70b-instruct:1.0.0
+  workspace: !workspace
+    components:
+      - dst: ''
+        src:
+          files:
+            - !name 'README.md'
+            - !name 'checksums.blake3'
+            - !name 'config.json'
+            - !name 'generation_config.json'
+            - !name 'merges.txt'
+            - !name 'model-00001-of-00001.safetensors'
+            - !name 'model.safetensors.index.json'
+            - !name 'params.json'
+            - !name 'special_tokens_map.json'
+            - !name 'tekken.json'
+            - !name 'tokenizer.json'
+            - !name 'tokenizer_config.json'
+            - !name 'vocab.json'
+          repo_id: 'ngc://nim/meta/llama3-70b-instruct:1.0.0'