From 12d54985ef7a1f4a03266902c2521568eeac7d88 Mon Sep 17 00:00:00 2001
From: ga1aXy-pog <87305200+ga1aXy-pog@users.noreply.github.com>
Date: Thu, 7 Jul 2022 11:52:30 +1000
Subject: [PATCH] Update benchmark.py

---
 benchmark/benchmark.py | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/benchmark/benchmark.py b/benchmark/benchmark.py
index 5d00e16f..674ea851 100644
--- a/benchmark/benchmark.py
+++ b/benchmark/benchmark.py
@@ -5,6 +5,7 @@
 import csv
 import os.path
 import math
+import random
 
 from qiskit import ClassicalRegister, QuantumRegister, QuantumCircuit
 from qiskit.wrapper import load_qasm_file
@@ -156,4 +157,10 @@ def benchmark(samples, qubits, out, single):
             write_csv(writer, {'name': func.__name__, 'num_qubits': n+1, 'time': t})
 
 if __name__ == '__main__':
-    benchmark()
\ No newline at end of file
+    benchmark()
+Finish = True;
+
+if Finish == True:
+    SystemExit();
+else:
+    random(SystemError);