From 26406ef892a3dea6cca5e92d4b379bef159640e2 Mon Sep 17 00:00:00 2001
From: Arjun Suresh <arjun@gateoverflow.com>
Date: Mon, 14 Apr 2025 00:21:59 +0100
Subject: [PATCH 1/5] Update customize.py

---
 script/get-ml-model-gptj/customize.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/script/get-ml-model-gptj/customize.py b/script/get-ml-model-gptj/customize.py
index cb15d7ce1..6dc97fc18 100644
--- a/script/get-ml-model-gptj/customize.py
+++ b/script/get-ml-model-gptj/customize.py
@@ -1,4 +1,5 @@
 from mlc import utils
+from utils import *
 import os
 
 

From 9c1aa3d1193af4e481a67c54dea5fa5a169316cb Mon Sep 17 00:00:00 2001
From: Arjun Suresh <arjun@gateoverflow.com>
Date: Mon, 14 Apr 2025 00:27:14 +0100
Subject: [PATCH 2/5] Delete automation/script/templates directory

---
 .../script/templates/default/customize.py     | 24 -------------------
 automation/script/templates/default/run.bat   |  1 -
 automation/script/templates/default/run.sh    | 17 -------------
 3 files changed, 42 deletions(-)
 delete mode 100644 automation/script/templates/default/customize.py
 delete mode 100644 automation/script/templates/default/run.bat
 delete mode 100644 automation/script/templates/default/run.sh

diff --git a/automation/script/templates/default/customize.py b/automation/script/templates/default/customize.py
deleted file mode 100644
index bd7c12dd3..000000000
--- a/automation/script/templates/default/customize.py
+++ /dev/null
@@ -1,24 +0,0 @@
-from cmind import utils
-import os
-
-
-def preprocess(i):
-
-    os_info = i['os_info']
-
-    env = i['env']
-
-    meta = i['meta']
-
-    automation = i['automation']
-
-    quiet = (env.get('MLC_QUIET', False) == 'yes')
-
-    return {'return': 0}
-
-
-def postprocess(i):
-
-    env = i['env']
-
-    return {'return': 0}
diff --git a/automation/script/templates/default/run.bat b/automation/script/templates/default/run.bat
deleted file mode 100644
index 648302ca7..000000000
--- a/automation/script/templates/default/run.bat
+++ /dev/null
@@ -1 +0,0 @@
-rem native script
diff --git a/automation/script/templates/default/run.sh b/automation/script/templates/default/run.sh
deleted file mode 100644
index 32cf4d51e..000000000
--- a/automation/script/templates/default/run.sh
+++ /dev/null
@@ -1,17 +0,0 @@
-#!/bin/bash
-
-#CM Script location: ${MLC_TMP_CURRENT_SCRIPT_PATH}
-
-#To export any variable
-#echo "VARIABLE_NAME=VARIABLE_VALUE" >>tmp-run-env.out
-
-#${MLC_PYTHON_BIN_WITH_PATH} contains the path to python binary if "get,python" is added as a dependency
-
-echo "Running: "
-echo "${MLC_RUN_CMD}"
-echo ""
-
-if [[ ${MLC_FAKE_RUN} != "yes" ]]; then
-  eval "${MLC_RUN_CMD}"
-  test $? -eq 0 || exit 1
-fi

From b2f6b2e60edf33116601260fafdb71d09ae24ea2 Mon Sep 17 00:00:00 2001
From: Arjun Suresh <arjun@gateoverflow.com>
Date: Mon, 14 Apr 2025 00:30:26 +0100
Subject: [PATCH 3/5] Update meta.yaml

---
 script/get-ml-model-gptj/meta.yaml | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/script/get-ml-model-gptj/meta.yaml b/script/get-ml-model-gptj/meta.yaml
index 2046cedc9..fa131dcf1 100644
--- a/script/get-ml-model-gptj/meta.yaml
+++ b/script/get-ml-model-gptj/meta.yaml
@@ -34,6 +34,12 @@ prehook_deps:
   update_tags_from_env_with_prefix:
     _url.:
     - MLC_DOWNLOAD_URL
+tests:
+  run_inputs:
+    - variations_list:
+        - nvidia,fp8
+  
+    
 print_env_at_the_end:
   MLC_ML_MODEL_FILE_WITH_PATH: Path to the ML model
 tags:

From 84a23077e9cba60f129d6431c24d554582a59b0c Mon Sep 17 00:00:00 2001
From: Arjun Suresh <arjun@gateoverflow.com>
Date: Mon, 14 Apr 2025 00:37:16 +0100
Subject: [PATCH 4/5] Update meta.yaml

---
 script/get-ml-model-gptj/meta.yaml | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/script/get-ml-model-gptj/meta.yaml b/script/get-ml-model-gptj/meta.yaml
index fa131dcf1..17d0c28a2 100644
--- a/script/get-ml-model-gptj/meta.yaml
+++ b/script/get-ml-model-gptj/meta.yaml
@@ -36,8 +36,7 @@ prehook_deps:
     - MLC_DOWNLOAD_URL
 tests:
   run_inputs:
-    - variations_list:
-        - nvidia,fp8
+    - fp32
   
     
 print_env_at_the_end:

From bff561e1aca85dd6a2b04103b7527e0a0f4eca91 Mon Sep 17 00:00:00 2001
From: Arjun Suresh <arjun@gateoverflow.com>
Date: Mon, 14 Apr 2025 00:37:35 +0100
Subject: [PATCH 5/5] Update meta.yaml

---
 script/get-ml-model-gptj/meta.yaml | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/script/get-ml-model-gptj/meta.yaml b/script/get-ml-model-gptj/meta.yaml
index 17d0c28a2..b54e3adeb 100644
--- a/script/get-ml-model-gptj/meta.yaml
+++ b/script/get-ml-model-gptj/meta.yaml
@@ -36,7 +36,8 @@ prehook_deps:
     - MLC_DOWNLOAD_URL
 tests:
   run_inputs:
-    - fp32
+    - variations_list:
+      - fp32
   
     
 print_env_at_the_end: