slide2vec 5.9.0__tar.gz → 5.9.1__tar.gz

{slide2vec-5.9.0 → slide2vec-5.9.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: slide2vec
-Version: 5.9.0
+Version: 5.9.1
 Summary: Embedding of whole slide images with Foundation Models
 Author-email: Clément Grisi <clement.grisi@radboudumc.nl>
 License-Expression: Apache-2.0

{slide2vec-5.9.0 → slide2vec-5.9.1}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "slide2vec"
-version = "5.9.0"
+version = "5.9.1"
 description = "Embedding of whole slide images with Foundation Models"
 readme = "README.md"
 requires-python = ">=3.10"
@@ -180,7 +180,7 @@ no_implicit_reexport = true
 max-line-length = 160
 
 [tool.bumpver]
-current_version = "5.9.0"
+current_version = "5.9.1"
 version_pattern = "MAJOR.MINOR.PATCH"
 commit = false
 tag = false

{slide2vec-5.9.0 → slide2vec-5.9.1}/slide2vec/__init__.py

@@ -27,7 +27,7 @@ from slide2vec.encoders import (
 )
 
 
-__version__ = "5.9.0"
+__version__ = "5.9.1"
 
 __all__ = [
     "Model",

{slide2vec-5.9.0 → slide2vec-5.9.1}/slide2vec/encoders/models/prism.py

@@ -46,5 +46,7 @@ class PrismSlideEncoder(SlideEncoder):
     ) -> torch.Tensor:
         if tile_features.ndim == 2:
             tile_features = tile_features.unsqueeze(0)
+        # Stored feature precision can differ from PRISM's weights; CPU has no autocast.
+        tile_features = tile_features.to(dtype=self._model.dtype)
         reprs = self._model.slide_representations(tile_features)
         return reprs["image_embedding"].squeeze(0)

{slide2vec-5.9.0 → slide2vec-5.9.1}/slide2vec.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: slide2vec
-Version: 5.9.0
+Version: 5.9.1
 Summary: Embedding of whole slide images with Foundation Models
 Author-email: Clément Grisi <clement.grisi@radboudumc.nl>
 License-Expression: Apache-2.0

{slide2vec-5.9.0 → slide2vec-5.9.1}/slide2vec.egg-info/SOURCES.txt

@@ -144,6 +144,7 @@ tests/test_patient_manifest.py
 tests/test_phaet.py
 tests/test_pooled_encoder_input.py
 tests/test_pooled_geometry.py
+tests/test_prism.py
 tests/test_prism2.py
 tests/test_progress.py
 tests/test_regression_core.py

slide2vec-5.9.1/tests/test_prism.py

@@ -0,0 +1,52 @@
+"""PRISM accepts persisted features independently of their storage dtype."""
+
+from types import SimpleNamespace
+
+import pytest
+
+torch = pytest.importorskip("torch")
+pytest.importorskip("transformers")
+
+
+class _TinyPrism(torch.nn.Module):
+    def __init__(self, dtype):
+        super().__init__()
+        self.weight = torch.nn.Parameter(
+            torch.tensor([[1.0, 2.0], [3.0, 4.0]], dtype=dtype)
+        )
+
+    @property
+    def dtype(self):
+        return self.weight.dtype
+
+    def slide_representations(self, tile_features):
+        projected = torch.nn.functional.linear(tile_features, self.weight)
+        return {"image_embedding": projected.sum(dim=1)}
+
+
+@pytest.mark.parametrize(
+    ("stored_dtype", "model_dtype"),
+    [
+        (torch.float16, torch.float32),
+        (torch.bfloat16, torch.float32),
+        (torch.float32, torch.float32),
+        (torch.float32, torch.float16),
+    ],
+)
+def test_prism_aggregates_features_in_model_dtype(monkeypatch, stored_dtype, model_dtype):
+    import slide2vec.encoders.models.prism as prism
+
+    monkeypatch.setattr(
+        prism,
+        "AutoModel",
+        SimpleNamespace(from_pretrained=lambda *args, **kwargs: _TinyPrism(model_dtype)),
+    )
+    encoder = prism.PrismSlideEncoder().to("cpu")
+    features = torch.tensor([[1.0, 2.0], [3.0, 4.0]], dtype=stored_dtype)
+
+    embedding = encoder.encode_slide(features)
+
+    torch.testing.assert_close(
+        embedding, torch.tensor([16.0, 36.0], dtype=model_dtype), rtol=0, atol=0,
+    )
+    assert features.dtype == stored_dtype