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