From 992d57f20a13bdc617d248a3fe57fb9efc92d8fc Mon Sep 17 00:00:00 2001
From: Napuh <55241721+Napuh@users.noreply.github.com>
Date: Thu, 19 Oct 2023 04:11:54 +0200
Subject: [PATCH] catch ConnectionError when checking dataset from HuggingFace
 (#743)

---
 src/axolotl/utils/data.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/axolotl/utils/data.py b/src/axolotl/utils/data.py
index c944dd27b..99697de32 100644
--- a/src/axolotl/utils/data.py
+++ b/src/axolotl/utils/data.py
@@ -158,7 +158,7 @@ def load_tokenized_prepared_datasets(
                     token=use_auth_token,
                 )
                 ds_from_hub = True
-            except FileNotFoundError:
+            except (FileNotFoundError, ConnectionError):
                 pass
 
             # prefer local dataset, even if hub exists