diff --git a/.riahub/workflows/workflow.yaml b/.riahub/workflows/workflow.yaml
index 2304f4f..e1cac93 100644
--- a/.riahub/workflows/workflow.yaml
+++ b/.riahub/workflows/workflow.yaml
@@ -37,12 +37,12 @@ jobs:
           python -m pip install --upgrade pip
           pip install -r requirements.txt
 
-      - name: 1. Build HDF5 Dataset
-        run: |
-         mkdir -p data/dataset
-         PYTHONPATH=. python data/scripts/produce_dataset.py
-         echo "datasets produced successfully"
-        shell: bash
+      # - name: 1. Build HDF5 Dataset
+      #   run: |
+      #    mkdir -p data/dataset
+      #    PYTHONPATH=. python data/scripts/produce_dataset.py
+      #    echo "datasets produced successfully"
+      #   shell: bash
       
       - name: Upload Dataset Artifacts
         uses: actions/upload-artifact@v3
diff --git a/data/training/train.py b/data/training/train.py
index f2106ab..4a2a3b6 100644
--- a/data/training/train.py
+++ b/data/training/train.py
@@ -85,7 +85,7 @@ def train_model():
     hparams = {
         "drop_path_rate": 0.2,
         "drop_rate": 0.5,
-        "learning_rate": training_cfg.learning_rate,
+        "learning_rate": float(training_cfg.learning_rate),
         "wd": 0.01,
     }