From 82f02ae9b15fbc564cd505a10fea803d857c5edb Mon Sep 17 00:00:00 2001
From: mrq <mrq@ecker.tech>
Date: Fri, 6 Oct 2023 09:26:52 -0500
Subject: [PATCH] oops

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

diff --git a/vall_e/data.py b/vall_e/data.py
index ea9a3a1..c3b3ede 100755
--- a/vall_e/data.py
+++ b/vall_e/data.py
@@ -115,7 +115,7 @@ def _get_hdf5_paths( data_dir, type="training", validate=False ):
 		duration = child.attrs['duration']
 		if type not in _total_durations:
 			_total_durations[type] = 0
-		_total_durations[type] += entry['duration']
+		_total_durations[type] += child.attrs['duration']
 		return cfg.dataset.min_duration <= duration and duration <= cfg.dataset.max_duration and cfg.dataset.min_phones <= phones and phones <= cfg.dataset.max_phones
 
 	key = f"/{type}{_get_hdf5_path(data_dir)}"