diff --git a/data.py b/data.py
index e9b238c..7df3348 100644
--- a/data.py
+++ b/data.py
@@ -9,6 +9,10 @@
 class Data:
     def __init__(self, dir_path):
         self.files = list(utils.find_files_by_extensions(dir_path, ['.pickle']))
+
+        if len(self.files) == 0:
+            raise ValueError("No pickle files found to analyze.")
+
         self.file_dict = {
             'train': self.files[:int(len(self.files) * 0.8)],
             'eval': self.files[int(len(self.files) * 0.8): int(len(self.files) * 0.9)],