From 1ba1b63eafdc5621c15bba63b367201282ec29dc Mon Sep 17 00:00:00 2001
From: ultimaxx <153694523+ultimaxx@users.noreply.github.com>
Date: Mon, 17 Feb 2025 03:44:14 -0600
Subject: [PATCH] progress bar fix (#8)

The ETA was broken when skipping frames
---
 auto-boost_2.5.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/auto-boost_2.5.py b/auto-boost_2.5.py
index bc1b33b..27cb119 100755
--- a/auto-boost_2.5.py
+++ b/auto-boost_2.5.py
@@ -200,7 +200,7 @@ def calculate_ssimu2(src_file, enc_file, ssimu2_txt_path, ranges, skip):
     with ssimu2_txt_path.open("w") as file:
         file.write(f"skip: {skip}\n")
     iter = 0
-    with tqdm(total=floor(len(source_clip) / int(skip)), desc=f'Calculating SSIMULACRA 2 scores') as pbar:
+    with tqdm(total=floor(len(source_clip)), desc=f'Calculating SSIMULACRA 2 scores') as pbar:
         for i in range(len(ranges) - 1):
             cut_source_clip = source_clip[ranges[i]:ranges[i+1]].std.SelectEvery(cycle=skip, offsets=1)
             cut_encoded_clip = encoded_clip[ranges[i]:ranges[i+1]].std.SelectEvery(cycle=skip, offsets=1)