Adding missing dependency

geomin12 · Feb 20, 2025 · f8be78c · f8be78c
1 parent 1f2169a
commit f8be78c
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 2 deletions.
diff --git a/sharktank_models/test_suite/benchmarks/run_benchmarks.py b/sharktank_models/test_suite/benchmarks/run_benchmarks.py
@@ -8,7 +8,7 @@
 import os
 from pathlib import Path
 import argparse
-
+import sys
 
 def main():
     parser = argparse.ArgumentParser()

diff --git a/sharktank_models/test_suite/regression_tests/run_thresholds.py b/sharktank_models/test_suite/regression_tests/run_thresholds.py
@@ -10,7 +10,6 @@
 import argparse
 import sys
 
-
 def main():
     parser = argparse.ArgumentParser()
     parser.add_argument("--model", type=str, default="sdxl")