Extract runtime output naming

Author: Donglai Wei

connectomics/decoding/tuning/optuna_tuner.py

@@ -40,6 +40,12 @@
     OPTUNA_AVAILABLE = False
 
 from connectomics.metrics.metrics_seg import adapted_rand
+from connectomics.runtime.output_naming import (
+    tta_cache_suffix,
+    tuning_best_params_filename,
+    tuning_best_params_filename_candidates,
+    tuning_study_db_filename,
+)
 
 from ..registry import get_decoder
 from ..utils import remove_small_instances
@@ -54,40 +60,6 @@
 ]
 
 
-def tta_cache_suffix(*args, **kwargs) -> str:
-    """Resolve the prediction cache suffix lazily to avoid import-time training coupling."""
-    from connectomics.training.lightning.utils import tta_cache_suffix as _tta_cache_suffix
-
-    return _tta_cache_suffix(*args, **kwargs)
-
-
-def tuning_best_params_filename(*args, **kwargs) -> str:
-    """Resolve the best-params filename lazily."""
-    from connectomics.training.lightning.utils import (
-        tuning_best_params_filename as _tuning_best_params_filename,
-    )
-
-    return _tuning_best_params_filename(*args, **kwargs)
-
-
-def tuning_best_params_filename_candidates(*args, **kwargs) -> list[str]:
-    """Resolve best-params filename candidates lazily."""
-    from connectomics.training.lightning.utils import (
-        tuning_best_params_filename_candidates as _tuning_best_params_filename_candidates,
-    )
-
-    return _tuning_best_params_filename_candidates(*args, **kwargs)
-
-
-def tuning_study_db_filename(*args, **kwargs) -> str:
-    """Resolve the Optuna study DB filename lazily."""
-    from connectomics.training.lightning.utils import (
-        tuning_study_db_filename as _tuning_study_db_filename,
-    )
-
-    return _tuning_study_db_filename(*args, **kwargs)
-
-
 class TrialEvaluationTimeoutError(TimeoutError):
     """Raised when a single tuning trial exceeds the configured wall-clock limit."""
 

connectomics/runtime/__init__.py

@@ -1,5 +1,38 @@
 """Runtime and CLI support helpers."""
 
+from .output_naming import (
+    compute_tta_passes,
+    final_prediction_output_tag,
+    format_checkpoint_name_tag,
+    format_decode_tag,
+    format_output_head_tag,
+    format_select_channel_tag,
+    is_tta_cache_suffix,
+    resolve_prediction_cache_suffix,
+    tta_cache_suffix,
+    tta_cache_suffix_candidates,
+    tuning_artifact_tag,
+    tuning_best_params_filename,
+    tuning_best_params_filename_candidates,
+    tuning_study_db_filename,
+)
 from .preflight import preflight_check, print_preflight_issues
 
-__all__ = ["preflight_check", "print_preflight_issues"]
+__all__ = [
+    "compute_tta_passes",
+    "final_prediction_output_tag",
+    "format_checkpoint_name_tag",
+    "format_decode_tag",
+    "format_output_head_tag",
+    "format_select_channel_tag",
+    "is_tta_cache_suffix",
+    "preflight_check",
+    "print_preflight_issues",
+    "resolve_prediction_cache_suffix",
+    "tta_cache_suffix",
+    "tta_cache_suffix_candidates",
+    "tuning_artifact_tag",
+    "tuning_best_params_filename",
+    "tuning_best_params_filename_candidates",
+    "tuning_study_db_filename",
+]