From 8075c4801afbf03782d81e663907d2e14be9e6ad Mon Sep 17 00:00:00 2001 From: Matthias Reso <13337103+mreso@users.noreply.github.com> Date: Mon, 14 Oct 2024 14:36:12 -0700 Subject: [PATCH] Remove debug print statements --- src/tests/test_finetuning.py | 2 -- src/tests/utils.py | 1 - 2 files changed, 3 deletions(-) diff --git a/src/tests/test_finetuning.py b/src/tests/test_finetuning.py index b6955775..9d9e6faa 100644 --- a/src/tests/test_finetuning.py +++ b/src/tests/test_finetuning.py @@ -246,8 +246,6 @@ def test_finetuning_weight_decay( args, kwargs = train.call_args optimizer = args[4] - print(optimizer.state_dict()) - assert isinstance(optimizer, AdamW) assert optimizer.state_dict()["param_groups"][0]["weight_decay"] == approx(0.01) diff --git a/src/tests/utils.py b/src/tests/utils.py index 9dbf3074..14b96a9a 100644 --- a/src/tests/utils.py +++ b/src/tests/utils.py @@ -20,7 +20,6 @@ class FakeTokenizer(object): self.padding_side = "left" def __call__(self, *args, **kwargs): - print(f"{kwargs=}") ids = self.encode(*args, **kwargs) return {"input_ids": ids} -- GitLab