diff --git a/src/flitrak3d/skeleton_fitter/modules/bodies/insect_body_3segments_2wings_hybrid.py b/src/flitrak3d/skeleton_fitter/modules/bodies/insect_body_3segments_2wings_hybrid.py index 8ade896db1a19e93aa7099fb271ac4a4a00dfd21..33747fee651ad8258426da62162987b8cfbda54c 100644 --- a/src/flitrak3d/skeleton_fitter/modules/bodies/insect_body_3segments_2wings_hybrid.py +++ b/src/flitrak3d/skeleton_fitter/modules/bodies/insect_body_3segments_2wings_hybrid.py @@ -11,6 +11,8 @@ from flitrak3d.skeleton_fitter.utils import set_axes3d_equal import flitrak3d.skeleton_fitter.modules.bodies.insect_body_3segments_2wings as body # Need these imports to be able to use these functions directly from these hybrid module (without duplicating them here) +from flitrak3d.skeleton_fitter.modules.bodies.insect_body_3segments_2wings import rotate_skeleton3d, translate_skeleton3d, \ + estimate_params_from_skeleton3d, get_copy_params_zero, residuals2d, residuals3d # noqa: F401 def init_params(init_dict: Dict[str, any]) -> Dict[str, float]: diff --git a/src/flitrak3d/skeleton_fitter/modules/limbs/insect_wings_flat_geo.py b/src/flitrak3d/skeleton_fitter/modules/limbs/insect_wings_flat_geo.py index ba8bf86e96c13666f3d290f1e66181b33de5c1df..a94f9cccc0bf248ab58378b7a3e3ea4a04e27433 100644 --- a/src/flitrak3d/skeleton_fitter/modules/limbs/insect_wings_flat_geo.py +++ b/src/flitrak3d/skeleton_fitter/modules/limbs/insect_wings_flat_geo.py @@ -11,8 +11,11 @@ from flitrak3d.skeleton_fitter.modules.bodies.insect_body_3segments_2wings impor from flitrak3d.skeleton_fitter.utils import reproject_skeleton3d_to2d -# Need these imports to be able to use these functions directly from insect_wings_flat (without duplicating them here) -from flitrak3d.skeleton_fitter.modules.limbs.insect_wings_flat import init_params, init_bounds, build_skeleton3d + # Need these imports to be able to use these functions directly from insect_wings_flat (without duplicating them here) +from flitrak3d.skeleton_fitter.modules.limbs.insect_wings_flat import init_params, init_bounds, build_skeleton3d, \ + rotate_and_translate_skeleton3d, translate_skeleton3d, estimate_params_from_skeleton3d, \ + postprocessing_estimate_limbs_params # noqa: F401 + # TODO remove commented code? # def init_wings_skeleton_geometry3d(wings_params):