Handle file_size passed as int

2021-07-21 06:23:19 -04:00 · 2021-07-21 06:23:19 -04:00 · f6bd7bcf7d
commit f6bd7bcf7d
parent 4dc565beca
1 changed files with 2 additions and 1 deletions
--- a/training/coqui_stt_training/util/helpers.py
+++ b/training/coqui_stt_training/util/helpers.py
@ -19,7 +19,8 @@ ValueRange = namedtuple("ValueRange", "start end r")


 def parse_file_size(file_size):
-    file_size = file_size.lower().strip()
+    if type(file_size) is str:
+        file_size = file_size.lower().strip()
    if len(file_size) == 0:
        return 0
    n = int(keep_only_digits(file_size))