update tests

kaiko-ai · Oct 9, 2024 · ccf34d9 · ccf34d9
1 parent 201e003
commit ccf34d9
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git 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
@@ -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: