nnInteractive 1.1.0__tar.gz → 1.1.2__tar.gz

{nninteractive-1.1.0 → nninteractive-1.1.2}/PKG-INFO

@@ -1,6 +1,6 @@
-Metadata-Version: 2.1
+Metadata-Version: 2.4
 Name: nnInteractive
-Version: 1.1.0
+Version: 1.1.2
 Summary: Inference code for nnInteractive
 Author: Helmholtz Imaging Applied Computer Vision Lab
 Author-email: Fabian Isensee <f.isensee@dkfz-heidelberg.de>
@@ -227,6 +227,7 @@ Provides-Extra: dev
 Requires-Dist: black; extra == "dev"
 Requires-Dist: ruff; extra == "dev"
 Requires-Dist: pre-commit; extra == "dev"
+Dynamic: license-file
 
 <img src="imgs/nnInteractive_header_white.png">
 
@@ -460,6 +461,16 @@ Note that while this repository is available under Apache-2.0 license (see [LICE
 
 # Changelog
 
+### 1.1.2 - 2025-08-02
+
+- Fixed a bug where `pin_memory` was set to `True` even though no CUDA devices were present (this broke CPU support)
+- ✅ API compatible all the way back to 1.0.1
+
+### 1.1.1 - 2025-08-01
+
+- We now detect whether linux kernel 6.11 is used and disable pin_memory in that case. See also [here](https://github.com/MIC-DKFZ/nnInteractive/issues/18)
+- ✅ API compatible with 1.0.1 and 1.1.0
+
 ### 1.1.0 - 2025-08-01
 
 - Reworked inference code. It's now well-structured and easier to follow.

nninteractive-1.1.2/nnInteractive/__init__.py

@@ -0,0 +1,3 @@
+from importlib.metadata import version as _version
+
+__version__ = _version("nnInteractive")

{nninteractive-1.1.0 → nninteractive-1.1.2}/nnInteractive/inference/inference_session.py

@@ -21,6 +21,7 @@ from nnInteractive.trainer.nnInteractiveTrainer import nnInteractiveTrainer_stub
 from nnInteractive.utils.bboxes import generate_bounding_boxes
 from nnInteractive.utils.crop import crop_and_pad_into_buffer, paste_tensor, pad_cropped, crop_to_valid
 from nnInteractive.utils.erosion_dilation import iterative_3x3_same_padding_pool3d
+from nnInteractive.utils.os_shennanigans import is_linux_kernel_6_11
 from nnInteractive.utils.rounding import round_to_nearest_odd
 
 
@@ -134,14 +135,16 @@ class nnInteractiveInferenceSession():
         self.has_positive_bbox = False
 
     def _initialize_interactions(self, image_torch: torch.Tensor):
+        # there is a bug in 6.11 that doesn't allow pinning large tensors
+        use_pinned = not is_linux_kernel_6_11() and self.use_pinned_memory and self.device.type == 'cuda'
         if self.verbose:
-            print(f'Initialize interactions. Pinned: {self.use_pinned_memory}')
+            print(f'Initialize interactions. Pinned: {use_pinned}')
         # Create the interaction tensor based on the target shape.
         self.interactions = torch.zeros(
             (7, *image_torch.shape[1:]),
             device='cpu',
             dtype=torch.float16,
-            pin_memory=False
+            pin_memory=use_pinned
         )
 
     def _background_set_image(self, image: np.ndarray, image_properties: dict):

nninteractive-1.1.2/nnInteractive/utils/os_shennanigans.py

@@ -0,0 +1,15 @@
+import platform
+import sys
+
+def is_linux_kernel_6_11():
+    if sys.platform != "linux":
+        return False
+
+    kernel_version = platform.release()  # e.g., '6.11.0-24-generic'
+    version_parts = kernel_version.split(".")
+    try:
+        major = int(version_parts[0])
+        minor = int(version_parts[1])
+        return major == 6 and minor == 11
+    except (IndexError, ValueError):
+        return False

{nninteractive-1.1.0 → nninteractive-1.1.2}/nnInteractive.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
-Metadata-Version: 2.1
+Metadata-Version: 2.4
 Name: nnInteractive
-Version: 1.1.0
+Version: 1.1.2
 Summary: Inference code for nnInteractive
 Author: Helmholtz Imaging Applied Computer Vision Lab
 Author-email: Fabian Isensee <f.isensee@dkfz-heidelberg.de>
@@ -227,6 +227,7 @@ Provides-Extra: dev
 Requires-Dist: black; extra == "dev"
 Requires-Dist: ruff; extra == "dev"
 Requires-Dist: pre-commit; extra == "dev"
+Dynamic: license-file
 
 <img src="imgs/nnInteractive_header_white.png">
 
@@ -460,6 +461,16 @@ Note that while this repository is available under Apache-2.0 license (see [LICE
 
 # Changelog
 
+### 1.1.2 - 2025-08-02
+
+- Fixed a bug where `pin_memory` was set to `True` even though no CUDA devices were present (this broke CPU support)
+- ✅ API compatible all the way back to 1.0.1
+
+### 1.1.1 - 2025-08-01
+
+- We now detect whether linux kernel 6.11 is used and disable pin_memory in that case. See also [here](https://github.com/MIC-DKFZ/nnInteractive/issues/18)
+- ✅ API compatible with 1.0.1 and 1.1.0
+
 ### 1.1.0 - 2025-08-01
 
 - Reworked inference code. It's now well-structured and easier to follow.

{nninteractive-1.1.0 → nninteractive-1.1.2}/nnInteractive.egg-info/SOURCES.txt

@@ -75,4 +75,5 @@ nnInteractive/utils/bboxes.py
 nnInteractive/utils/checkpoint_cleansing.py
 nnInteractive/utils/crop.py
 nnInteractive/utils/erosion_dilation.py
+nnInteractive/utils/os_shennanigans.py
 nnInteractive/utils/rounding.py

{nninteractive-1.1.0 → nninteractive-1.1.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "nnInteractive"
-version = "1.1.0"
+version = "1.1.2"
 requires-python = ">=3.10"
 description = "Inference code for nnInteractive"
 readme = "readme.md"

{nninteractive-1.1.0 → nninteractive-1.1.2}/readme.md

@@ -230,6 +230,16 @@ Note that while this repository is available under Apache-2.0 license (see [LICE
 
 # Changelog
 
+### 1.1.2 - 2025-08-02
+
+- Fixed a bug where `pin_memory` was set to `True` even though no CUDA devices were present (this broke CPU support)
+- ✅ API compatible all the way back to 1.0.1
+
+### 1.1.1 - 2025-08-01
+
+- We now detect whether linux kernel 6.11 is used and disable pin_memory in that case. See also [here](https://github.com/MIC-DKFZ/nnInteractive/issues/18)
+- ✅ API compatible with 1.0.1 and 1.1.0
+
 ### 1.1.0 - 2025-08-01
 
 - Reworked inference code. It's now well-structured and easier to follow.

nninteractive-1.1.0/nnInteractive/__init__.py

@@ -1 +0,0 @@
-__version__ = "1.1.0"