From 02dc87f8c1af13007db67aa0fc812b00c7cf9a10 Mon Sep 17 00:00:00 2001
From: Wing Lian <wing.lian@gmail.com>
Date: Fri, 5 Jan 2024 11:22:34 -0500
Subject: [PATCH] fix dict check

---
 src/axolotl/utils/data.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/src/axolotl/utils/data.py b/src/axolotl/utils/data.py
index 0de5d4ee32..40a3306021 100644
--- a/src/axolotl/utils/data.py
+++ b/src/axolotl/utils/data.py
@@ -71,9 +71,9 @@ def prepare_dataset(cfg, tokenizer):
     else:
         path = cfg.pretraining_dataset
         name = None
-        if isinstance(dict, cfg.pretraining_dataset):
-            path = cfg.pretraining_dataset.path
-            name = cfg.pretraining_dataset.name
+        if isinstance(cfg.pretraining_dataset, dict):
+            path = cfg.pretraining_dataset["path"]
+            name = cfg.pretraining_dataset["name"]
 
         train_dataset = load_pretraining_dataset(
             path,