Skip to content

Commit 9bd6e49

Browse files
committed
filter these warnings
1 parent 8dd6605 commit 9bd6e49

2 files changed

Lines changed: 8 additions & 1 deletion

File tree

codeflash/tracing/replay_test.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -44,7 +44,10 @@ def get_function_alias(module: str, function_name: str) -> str:
4444

4545

4646
def create_trace_replay_test(trace_file: str, functions: list[FunctionModules], max_run_count: int = 100) -> str:
47-
imports = """import dill as pickle
47+
imports = """import warnings
48+
import dill as pickle
49+
from dill import PicklingWarning
50+
warnings.filterwarnings("ignore", category=PicklingWarning)
4851
from codeflash.tracing.replay_test import get_next_arg_and_return
4952
"""
5053

codeflash/verification/codeflash_capture.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -7,11 +7,15 @@
77
import os
88
import sqlite3
99
import time
10+
import warnings
1011
from enum import Enum
1112
from pathlib import Path
1213
from typing import Callable
1314

1415
import dill as pickle
16+
from dill import PicklingWarning
17+
18+
warnings.filterwarnings("ignore", category=PicklingWarning)
1519

1620

1721
class VerificationType(str, Enum):

0 commit comments

Comments
 (0)