fix: delete pointer more properly

seung-lab · Dec 15, 2023 · 34b5cd0 · 34b5cd0
1 parent 2182a0c
commit 34b5cd0
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/fastmorph/fastmorphops.cpp b/fastmorph/fastmorphops.cpp
@@ -134,9 +134,10 @@ py::array dilate_helper(
 		}
 	}
 
-
 	py::capsule capsule(output, [](void* ptr) {
-		delete[] ptr;
+	    if (ptr) {
+	        delete[] static_cast<LABEL*>(ptr);
+	    }
 	});
 
 	uint64_t width = sizeof(LABEL);