fix: prepare_data_per_node is a flag and was incorrectly used as a re…

…placement for prepare_data
IGNF · Apr 18, 2024 · d52810e · d52810e
1 parent 8c01bdf
commit d52810e
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 2 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -1,4 +1,7 @@
 # CHANGELOG
+### 3.8.3
+- fix: prepare_data_per_node is a flag and was incorrectly used as a replacement for prepare_data.
+
 ### 3.8.2
 - fix: points not dropped case in subsampling when the subtile contains only one point
 - fix: type error in edge case when dropping points in DropPointsByClass (when there is only one remaining point)

diff --git a/myria3d/pctl/datamodule/hdf5.py b/myria3d/pctl/datamodule/hdf5.py
@@ -88,7 +88,7 @@ def eval_transform(self) -> CustomCompose:
     def predict_transform(self) -> CustomCompose:
         return CustomCompose(self.preparation_predict_transform + self.normalization_transform)
 
-    def prepare_data_per_node(self, stage: Optional[str] = None):
+    def prepare_data(self, stage: Optional[str] = None):
         """Prepare dataset containing train, val, test data."""
 
         if stage in ["fit", "test"] or stage is None:

diff --git a/package_metadata.yaml b/package_metadata.yaml
@@ -1,4 +1,4 @@
-__version__: "3.8.2"
+__version__: "3.8.3"
 __name__: "myria3d"
 __url__: "https://github.com/IGNF/myria3d"
 __description__: "Deep Learning for the Semantic Segmentation of Aerial Lidar Point Clouds"