get terminal size from shutil instead

mlcommons · Nov 13, 2024 · 90e779d · 90e779d
1 parent b7faf5c
commit 90e779d
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/modelbench/consistency_checker.py b/src/modelbench/consistency_checker.py
@@ -1,5 +1,5 @@
 import json
-import os
+import shutil
 from abc import ABC, abstractmethod
 from collections import Counter, defaultdict
 from itertools import product
@@ -8,7 +8,7 @@
 
 from modelbench.run_journal import journal_reader
 
-LINE_WIDTH = os.get_terminal_size().columns
+LINE_WIDTH = shutil.get_terminal_size(fallback=(120, 50)).columns
 
 
 class JournalSearch:

diff --git a/tests/modelbench_tests/test_consistency_checker.py b/tests/modelbench_tests/test_consistency_checker.py