add .clone()

aramis-lab · Dec 11, 2024 · 7e711bc · 7e711bc
1 parent 8f1ba86
commit 7e711bc
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 4 deletions.
diff --git a/clinicadl/transforms/extraction/image.py b/clinicadl/transforms/extraction/image.py
@@ -112,7 +112,7 @@ def extract_sample(
                 "1 sample in the image."
             )
 
-        return image_tensor
+        return image_tensor.clone()
 
     def sample_path(self, image_path: Path, sample_index: int = 0) -> Path:  # pylint:disable=unused-argument
         """

diff --git a/clinicadl/transforms/extraction/patch.py b/clinicadl/transforms/extraction/patch.py
@@ -167,9 +167,8 @@ def extract_sample(
         This method allows for the extraction of individual patches based on the provided index.
         """
         patches_tensor = self.get_patches(image_tensor)
-        print(patches_tensor[1])
         try:
-            return patches_tensor[sample_index, ...].unsqueeze_(0).clone()
+            return patches_tensor[sample_index].unsqueeze(0).clone()
         except IndexError as exc:
             raise IndexError(
                 f"'sample_index' {sample_index} is out of range as there are only "

diff --git a/clinicadl/transforms/extraction/roi.py b/clinicadl/transforms/extraction/roi.py
@@ -264,7 +264,7 @@ def extract_sample(
                 :, x_min : x_max + 1, y_min : y_max + 1, z_min : z_max + 1
             ]
 
-        return roi_tensor.float().clone()
+        return roi_tensor.clone()
 
     # TODO : to revise -> roi filename is not unique
     def sample_path(self, image_path: Path, sample_index: int) -> Path: