Consolidate multisession assignment and ROI similarity utilities

src/pyrecest/utils/__init__.py

@@ -1,21 +1,51 @@
+"""Utility helpers for :mod:`pyrecest`."""
 
 from .association_models import LogisticPairwiseAssociationModel
 from .history_recorder import HistoryRecorder
 from .assignment import murty_k_best_assignments
-
+from .multisession_assignment import (
+    MultiSessionAssignmentResult,
+    solve_multisession_assignment,
+    solve_multisession_assignment_from_similarity,
+    tracks_to_index_matrix,
+    tracks_to_session_labels,
+)
+from .multisession_assignment_observation_costs import (
+    solve_multisession_assignment_with_observation_costs,
+)
 from .nonrigid_point_set_registration import (
     ThinPlateSplineRegistrationResult,
     ThinPlateSplineTransform,
     estimate_thin_plate_spline,
     joint_tps_registration_assignment,
 )
+from .roi_similarity import (
+    build_roi_cost_matrix,
+    pairwise_centroid_distances,
+    pairwise_roi_similarity,
+    roi_centroid,
+    weighted_roi_cosine_similarity,
+    weighted_roi_jaccard,
+)
 
 __all__ = [
     "LogisticPairwiseAssociationModel",
+    "MultiSessionAssignmentResult",
     "HistoryRecorder",
     "ThinPlateSplineRegistrationResult",
     "ThinPlateSplineTransform",
+    "build_roi_cost_matrix",
     "estimate_thin_plate_spline",
     "joint_tps_registration_assignment",
     "murty_k_best_assignments",
-]
+    "pairwise_centroid_distances",
+    "pairwise_roi_similarity",
+    "roi_centroid",
+    "solve_multisession_assignment",
+    "solve_multisession_assignment_from_similarity",
+    "solve_multisession_assignment_with_observation_costs",
+    "tracks_to_index_matrix",
+    "tracks_to_session_labels",
+    "weighted_roi_cosine_similarity",
+    "weighted_roi_jaccard",
+]