Fix syaz0 header bug with RSTB estimates

NiceneNerd · Jan 14, 2020 · bd360f2 · bd360f2
1 parent 4e3aae8
commit bd360f2
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/bcml/rstable.py b/bcml/rstable.py
@@ -86,7 +86,7 @@ def guess_bfres_size(file: Union[Path, bytes], name: str = '') -> int:
     else:
         if file.suffix.startswith('.s'):
             with file.open('rb') as f:
-                real_size = syaz0.get_header(f.read(16))
+                real_size = syaz0.get_header(f.read(16)).uncompressed_size
         else:
             real_size = file.stat().st_size
     if name == '':
@@ -161,7 +161,7 @@ def guess_aamp_size(file: Union[Path, bytes], ext: str = '') -> int:
     else:
         if file.suffix.startswith('.s'):
             with file.open('rb') as f:
-                real_size = syaz0.get_header(f.read(16))
+                real_size = syaz0.get_header(f.read(16)).uncompressed_size
         else:
             real_size = file.stat().st_size
     if ext == '':