From ccf34d9d1d642f5740ee3a1ce68dd91acece52eb Mon Sep 17 00:00:00 2001 From: Nicolas Kaenzig Date: Wed, 9 Oct 2024 09:39:36 +0000 Subject: [PATCH] update tests --- configs/vision/tests/offline/panda.yaml | 2 +- .../core/callbacks/writers/embeddings/test_classification.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/configs/vision/tests/offline/panda.yaml b/configs/vision/tests/offline/panda.yaml index 28844dd1e..e5d5ecd15 100644 --- a/configs/vision/tests/offline/panda.yaml +++ b/configs/vision/tests/offline/panda.yaml @@ -14,7 +14,7 @@ trainer: 0: train 1: val 2: test - metadata_keys: ["wsi_id"] + metadata_keys: ["wsi_id", "x", "y", "width", "height", "level_idx"] backbone: class_path: eva.models.ModelFromFunction init_args: diff --git a/tests/eva/core/callbacks/writers/embeddings/test_classification.py b/tests/eva/core/callbacks/writers/embeddings/test_classification.py index 55488fdbb..c88e0a3f4 100644 --- a/tests/eva/core/callbacks/writers/embeddings/test_classification.py +++ b/tests/eva/core/callbacks/writers/embeddings/test_classification.py @@ -29,7 +29,7 @@ (5, 7, None, None), (5, 7, ["wsi_id"], None), (8, 16, None, None), - (8, 32, ["wsi_id"], ["slide_1", "slide_2"]), + (8, 32, ["wsi_id", "x", "y"], ["slide_1", "slide_2"]), ], ) def test_embeddings_writer(datamodule: datamodules.DataModule, model: modules.HeadModule) -> None: