Simplify benchmarks.py assertions (#8515)

ultralytics · Jul 7, 2022 · f17444a · f17444a
1 parent 3e54651
commit f17444a
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/utils/benchmarks.py b/utils/benchmarks.py
@@ -61,12 +61,12 @@ def run(
     device = select_device(device)
     for i, (name, f, suffix, cpu, gpu) in export.export_formats().iterrows():  # index, (name, file, suffix, CPU, GPU)
         try:
-            assert i not in (9, 10), f'{name} inference not supported'  # Edge TPU and TF.js are unsupported
-            assert i != 5 or platform.system() == 'Darwin', f'{name} inference only supported on macOS>=10.13'
+            assert i not in (9, 10), 'inference not supported'  # Edge TPU and TF.js are unsupported
+            assert i != 5 or platform.system() == 'Darwin', 'inference only supported on macOS>=10.13'  # CoreML
             if 'cpu' in device.type:
-                assert cpu, f'{name} inference not supported on CPU'
+                assert cpu, 'inference not supported on CPU'
             if 'cuda' in device.type:
-                assert gpu, f'{name} inference not supported on GPU'
+                assert gpu, 'inference not supported on GPU'
 
             # Export
             if f == '-':