@@ -139,7 +139,42 @@ def generate_unique_ids(min: int, max: int, n: int, trials: int = 20) -> np.arra
139139 return ids
140140
141141
142- def get_auto_lims (probe , margin = 40 ):
142+ def get_auto_lims (probe : Probe , margin : float = 40.0 ) -> tuple [float , float , float ]:
143+ """
144+ Automatically compute a probe boundaries given its contour and an optional
145+ margin. Both planar and tridimensional probes are accepted.
146+
147+ Parameters
148+ ----------
149+ probe : Probe
150+ The probe whose limits are to be computed.
151+ margin : float, default: 40
152+ An isotropic margin added to the exact probe boundaries.
153+
154+ Returns
155+ -------
156+ lims : a tuple containing xlims, ylims, and zlims. If the provided probe
157+ is planar, then zlims is None.
158+
159+ """
160+
161+ """
162+ Compute the boundaries of a given probe, considering its contour and an optional margin.
163+ The function is designed to handle both planar and tridimensional probes.
164+
165+ Parameters
166+ ----------
167+ probe : Probe
168+ The probe for which the limits are to be computed.
169+ margin : float, default: 40
170+ An isotropic margin that is added to the exact probe boundaries.
171+
172+ Returns
173+ -------
174+ lims : a tuple containing the limits in the x, y, and z directions
175+ (xlims, ylims, zlims). If the provided probe is planar, then
176+ zlims is None.
177+ """
143178 positions = probe .contact_positions
144179 planar_contour = probe .probe_planar_contour
145180
0 commit comments