From dce1ef77bb39cd092e68437b1edee32902645319 Mon Sep 17 00:00:00 2001
From: Bao Bui <bao.bui@shopee.com>
Date: Sat, 22 Apr 2023 11:36:33 +0800
Subject: [PATCH] Fix loading dataset

---
 datasets.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/datasets.py b/datasets.py
index 443c786..33148d6 100644
--- a/datasets.py
+++ b/datasets.py
@@ -46,7 +46,7 @@ def __init__(
             dataset_archive = DEFAULT_DATA_ARCHIVE
 
         data_path = tf.keras.utils.get_file(
-            origin=DEFAULT_DATA_ARCHIVE,
+            origin=dataset_archive,
             untar=True,
         )