diff --git a/v1/v1.py b/v1/v1.py index f40fd97..42188ae 100644 --- a/v1/v1.py +++ b/v1/v1.py @@ -28,6 +28,7 @@ class NVCCPlugin(Magics): stmt = f"subprocess.check_output(['{file_path}.out'], stderr=subprocess.STDOUT)" output = self.shell.run_cell_magic( magic_name="timeit", line="-q -o import subprocess", cell=stmt) + output = str(output) # convert TimeitResult object to human readable string else: output = subprocess.check_output( [file_path + ".out"], stderr=subprocess.STDOUT)