From b32ebbe502af6c45d3b0850a5aa7d22884fa8be4 Mon Sep 17 00:00:00 2001
From: dcarron <daniel.carron@idiap.ch>
Date: Tue, 30 Apr 2024 13:25:15 +0200
Subject: [PATCH] [tests] Update cli tests
---
tests/test_cli.py | 6 ++----
1 file changed, 2 insertions(+), 4 deletions(-)
diff --git a/tests/test_cli.py b/tests/test_cli.py
index 4ee0e2c6..a512d6ad 100644
--- a/tests/test_cli.py
+++ b/tests/test_cli.py
@@ -241,8 +241,7 @@ def test_train_pasa_montgomery(temporary_basedir):
keywords = {
r"^Loading dataset:`train` without caching. Trade-off: CPU RAM usage: less | Disk I/O: more.$": 1,
r"^Loading dataset:`validation` without caching. Trade-off: CPU RAM usage: less | Disk I/O: more.$": 1,
- r"^Applying DataModule train sampler balancing...$": 1,
- r"^Balancing samples from dataset using metadata targets `label`$": 1,
+ r"^Applying train/valid loss balancing...$": 1,
r"^Training for at most 1 epochs.$": 1,
r"^Uninitialised pasa model - computing z-norm factors from train dataloader.$": 1,
r"^Writing run metadata at.*$": 1,
@@ -323,8 +322,7 @@ def test_train_pasa_montgomery_from_checkpoint(temporary_basedir):
keywords = {
r"^Loading dataset:`train` without caching. Trade-off: CPU RAM usage: less | Disk I/O: more.$": 1,
r"^Loading dataset:`validation` without caching. Trade-off: CPU RAM usage: less | Disk I/O: more.$": 1,
- r"^Applying DataModule train sampler balancing...$": 1,
- r"^Balancing samples from dataset using metadata targets `label`$": 1,
+ r"^Applying train/valid loss balancing...$": 1,
r"^Training for at most 2 epochs.$": 1,
r"^Resuming from epoch 0 \(checkpoint file: .*$": 1,
r"^Writing run metadata at.*$": 1,
--
GitLab