Browse Source

Remove debug print statements

Matthias Reso 7 months ago
parent
commit
8075c4801a
2 changed files with 0 additions and 3 deletions
  1. 0 2
      src/tests/test_finetuning.py
  2. 0 1
      src/tests/utils.py

+ 0 - 2
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)
 

+ 0 - 1
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}