diff --git a/geoh5py/objects/drillhole.py b/geoh5py/objects/drillhole.py index f82bf0df..4e7bddab 100644 --- a/geoh5py/objects/drillhole.py +++ b/geoh5py/objects/drillhole.py @@ -811,12 +811,10 @@ def compute_intervals(survey: np.ndarray, collar, end_of_hole) -> dict: norm[norm == 0.0] = INFINITE_RADIUS tangential /= norm[:, None] alpha = np.abs(0.5 * np.pi - np.arctan2(dot, vr)) + alpha[alpha == 0.0] = INFINITE_RADIUS**-1.0 delta_depth = np.diff(full_survey[:, 0]) radius = delta_depth / alpha - radius[alpha == 0.0] = delta_depth[alpha == 0.0] * INFINITE_RADIUS - alpha[alpha == 0.0] = delta_depth[alpha == 0.0] / radius[alpha == 0.0] - intervals = { "depths": np.r_[full_survey[:, 0]], "rad": np.r_[radius, INFINITE_RADIUS],