Skip to content

Commit da6385f

Browse files
committed
use dill instead of pickle
1 parent 77f43a5 commit da6385f

File tree

1 file changed

+1
-3
lines changed

1 file changed

+1
-3
lines changed

codeflash/benchmarking/codeflash_trace.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
import functools
22
import os
3-
import pickle
3+
import dill as pickle
44
import time
55
from typing import Callable
66

@@ -63,13 +63,11 @@ def wrapper(*args, **kwargs):
6363
overhead_end_time = time.thread_time_ns()
6464
overhead_time = overhead_end_time - overhead_start_time
6565

66-
6766
self.function_calls_data.append(
6867
(func.__name__, class_name, func.__module__, func.__code__.co_filename,
6968
benchmark_function_name, benchmark_file_name, benchmark_line_number, execution_time,
7069
overhead_time, pickled_args, pickled_kwargs)
7170
)
72-
print("appended")
7371
except Exception as e:
7472
print(f"Error in codeflash_trace: {e}")
7573

0 commit comments

Comments
 (0)