Skip to content

Commit f8cbcbd

Browse files
committed
replay test template bug
1 parent b1d0218 commit f8cbcbd

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

codeflash/benchmarking/replay_test.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -116,7 +116,7 @@ def create_trace_replay_test_code(
116116
ret = {class_name_alias}(*args[1:], **kwargs)
117117
else:
118118
instance = args[0] # self
119-
ret = instance{method_name}(*args[1:], **kwargs)
119+
ret = {class_name_alias}{method_name}(*args, **kwargs)
120120
"""
121121
)
122122

@@ -127,7 +127,7 @@ def create_trace_replay_test_code(
127127
kwargs = pickle.loads(kwargs_pkl){filter_variables}
128128
if not args:
129129
raise ValueError("No arguments provided for the method.")
130-
ret = {class_name_alias}{method_name}(*args, **kwargs)
130+
ret = {class_name_alias}{method_name}(*args[1:], **kwargs)
131131
"""
132132
)
133133
test_static_method_body = textwrap.dedent(

0 commit comments

Comments
 (0)