From 3f738ab4756499b5e956e03fad2fc42e8bd9406e Mon Sep 17 00:00:00 2001
From: ANANDHU S <71482562+anandhu-eng@users.noreply.github.com>
Date: Tue, 4 Mar 2025 09:40:38 +0530
Subject: [PATCH 1/2] add num of workers

---
 script/get-dataset-coco2014/meta.yaml | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/script/get-dataset-coco2014/meta.yaml b/script/get-dataset-coco2014/meta.yaml
index fa3724f83..afdff9cc4 100644
--- a/script/get-dataset-coco2014/meta.yaml
+++ b/script/get-dataset-coco2014/meta.yaml
@@ -9,6 +9,9 @@ cache: true
 category: AI/ML datasets
 category_sort: 8500
 
+input_mapping:
+  num_workers: MLC_DATASET_COCO2014_NUM_WORKERS
+
 tags:
 - get
 - dataset
@@ -18,7 +21,8 @@ tags:
 
 default_env:
   MLC_DATASET_CALIBRATION: 'no'
-
+  MLC_DATASET_COCO2014_NUM_WORKERS: 1
+  
 deps:
 
 - names:

From adfdea84ae9ebd56acafdb9b89526efaeb8227c4 Mon Sep 17 00:00:00 2001
From: ANANDHU S <71482562+anandhu-eng@users.noreply.github.com>
Date: Tue, 4 Mar 2025 09:41:13 +0530
Subject: [PATCH 2/2] Update run.sh

---
 script/get-dataset-coco2014/run.sh | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/script/get-dataset-coco2014/run.sh b/script/get-dataset-coco2014/run.sh
index a891b2330..21829568e 100644
--- a/script/get-dataset-coco2014/run.sh
+++ b/script/get-dataset-coco2014/run.sh
@@ -27,7 +27,7 @@ if [[ ${MLC_DATASET_CALIBRATION} == "no" ]]; then
   eval $cmd
   test $? -eq 0 || exit $?
 else
-  cmd="./download-coco-2014-calibration.sh -d ${INSTALL_DIR}"
+  cmd="./download-coco-2014-calibration.sh -d ${INSTALL_DIR} -n ${MLC_DATASET_COCO2014_NUM_WORKERS}"
   echo $cmd
   eval $cmd
   test $? -eq 0 || exit $?