Skip to content

Commit 8075c48

Browse files
committed
Remove debug print statements
1 parent 448af9d commit 8075c48

File tree

2 files changed

+0
-3
lines changed

2 files changed

+0
-3
lines changed

src/tests/test_finetuning.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -246,8 +246,6 @@ def test_finetuning_weight_decay(
246246
args, kwargs = train.call_args
247247
optimizer = args[4]
248248

249-
print(optimizer.state_dict())
250-
251249
assert isinstance(optimizer, AdamW)
252250
assert optimizer.state_dict()["param_groups"][0]["weight_decay"] == approx(0.01)
253251

src/tests/utils.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,6 @@ def __init__(self):
2020
self.padding_side = "left"
2121

2222
def __call__(self, *args, **kwargs):
23-
print(f"{kwargs=}")
2423
ids = self.encode(*args, **kwargs)
2524
return {"input_ids": ids}
2625

0 commit comments

Comments
 (0)