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 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 diff --git a/script/get-ml-model-gptj/meta.yaml b/script/get-ml-model-gptj/meta.yaml index 2046cedc9..b54e3adeb 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: + - fp32 + + print_env_at_the_end: MLC_ML_MODEL_FILE_WITH_PATH: Path to the ML model tags: