Make pre_vector_to_weighting and pref_vector_to_str_suffix public

Author: PierreQuinton

src/torchjd/aggregation/_pref_vector_utils.py

@@ -5,7 +5,7 @@
 from .constant import _ConstantWeighting
 
 
-def _pref_vector_to_weighting(pref_vector: Tensor | None, default: _Weighting) -> _Weighting:
+def pref_vector_to_weighting(pref_vector: Tensor | None, default: _Weighting) -> _Weighting:
     """
     Returns the weighting associated to a given preference vector, with a fallback to a default
     weighting if the preference vector is None.
@@ -22,7 +22,7 @@ def _pref_vector_to_weighting(pref_vector: Tensor | None, default: _Weighting) -
         return _ConstantWeighting(pref_vector)
 
 
-def _pref_vector_to_str_suffix(pref_vector: Tensor | None) -> str:
+def pref_vector_to_str_suffix(pref_vector: Tensor | None) -> str:
     """Returns a suffix string containing the representation of the optional preference vector."""
 
     if pref_vector is None:

src/torchjd/aggregation/aligned_mtl.py

@@ -28,7 +28,7 @@
 import torch
 from torch import Tensor
 
-from ._pref_vector_utils import _pref_vector_to_str_suffix, _pref_vector_to_weighting
+from ._pref_vector_utils import pref_vector_to_str_suffix, pref_vector_to_weighting
 from .bases import _WeightedAggregator, _Weighting
 from .mean import _MeanWeighting
 
@@ -61,7 +61,7 @@ class AlignedMTL(_WeightedAggregator):
     """
 
     def __init__(self, pref_vector: Tensor | None = None):
-        weighting = _pref_vector_to_weighting(pref_vector, default=_MeanWeighting())
+        weighting = pref_vector_to_weighting(pref_vector, default=_MeanWeighting())
         self._pref_vector = pref_vector
 
         super().__init__(weighting=_AlignedMTLWrapper(weighting))
@@ -70,7 +70,7 @@ def __repr__(self) -> str:
         return f"{self.__class__.__name__}(pref_vector={repr(self._pref_vector)})"
 
     def __str__(self) -> str:
-        return f"AlignedMTL{_pref_vector_to_str_suffix(self._pref_vector)}"
+        return f"AlignedMTL{pref_vector_to_str_suffix(self._pref_vector)}"
 
 
 class _AlignedMTLWrapper(_Weighting):

src/torchjd/aggregation/config.py

@@ -28,7 +28,7 @@
 import torch
 from torch import Tensor
 
-from ._pref_vector_utils import _pref_vector_to_str_suffix, _pref_vector_to_weighting
+from ._pref_vector_utils import pref_vector_to_str_suffix, pref_vector_to_weighting
 from .bases import Aggregator
 from .sum import _SumWeighting
 
@@ -62,7 +62,7 @@ class ConFIG(Aggregator):
 
     def __init__(self, pref_vector: Tensor | None = None):
         super().__init__()
-        self.weighting = _pref_vector_to_weighting(pref_vector, default=_SumWeighting())
+        self.weighting = pref_vector_to_weighting(pref_vector, default=_SumWeighting())
         self._pref_vector = pref_vector
 
     def forward(self, matrix: Tensor) -> Tensor:
@@ -80,4 +80,4 @@ def __repr__(self) -> str:
         return f"{self.__class__.__name__}(pref_vector={repr(self._pref_vector)})"
 
     def __str__(self) -> str:
-        return f"ConFIG{_pref_vector_to_str_suffix(self._pref_vector)}"
+        return f"ConFIG{pref_vector_to_str_suffix(self._pref_vector)}"

src/torchjd/aggregation/dualproj.py

@@ -4,7 +4,7 @@
 
 from ._dual_cone_utils import project_weights
 from ._gramian_utils import _compute_regularized_normalized_gramian
-from ._pref_vector_utils import _pref_vector_to_str_suffix, _pref_vector_to_weighting
+from ._pref_vector_utils import pref_vector_to_str_suffix, pref_vector_to_weighting
 from .bases import _WeightedAggregator, _Weighting
 from .mean import _MeanWeighting
 
@@ -47,7 +47,7 @@ def __init__(
         reg_eps: float = 0.0001,
         solver: Literal["quadprog"] = "quadprog",
     ):
-        weighting = _pref_vector_to_weighting(pref_vector, default=_MeanWeighting())
+        weighting = pref_vector_to_weighting(pref_vector, default=_MeanWeighting())
         self._pref_vector = pref_vector
 
         super().__init__(
@@ -64,7 +64,7 @@ def __repr__(self) -> str:
         )
 
     def __str__(self) -> str:
-        return f"DualProj{_pref_vector_to_str_suffix(self._pref_vector)}"
+        return f"DualProj{pref_vector_to_str_suffix(self._pref_vector)}"
 
 
 class _DualProjWrapper(_Weighting):

src/torchjd/aggregation/upgrad.py

@@ -5,7 +5,7 @@
 
 from ._dual_cone_utils import project_weights
 from ._gramian_utils import _compute_regularized_normalized_gramian
-from ._pref_vector_utils import _pref_vector_to_str_suffix, _pref_vector_to_weighting
+from ._pref_vector_utils import pref_vector_to_str_suffix, pref_vector_to_weighting
 from .bases import _WeightedAggregator, _Weighting
 from .mean import _MeanWeighting
 
@@ -47,7 +47,7 @@ def __init__(
         reg_eps: float = 0.0001,
         solver: Literal["quadprog"] = "quadprog",
     ):
-        weighting = _pref_vector_to_weighting(pref_vector, default=_MeanWeighting())
+        weighting = pref_vector_to_weighting(pref_vector, default=_MeanWeighting())
         self._pref_vector = pref_vector
 
         super().__init__(
@@ -64,7 +64,7 @@ def __repr__(self) -> str:
         )
 
     def __str__(self) -> str:
-        return f"UPGrad{_pref_vector_to_str_suffix(self._pref_vector)}"
+        return f"UPGrad{pref_vector_to_str_suffix(self._pref_vector)}"
 
 
 class _UPGradWrapper(_Weighting):

tests/unit/aggregation/test_pref_vector_utils.py

@@ -5,7 +5,7 @@
 from torch import Tensor
 from unit._utils import ExceptionContext
 
-from torchjd.aggregation._pref_vector_utils import _pref_vector_to_weighting
+from torchjd.aggregation._pref_vector_utils import pref_vector_to_weighting
 from torchjd.aggregation.mean import _MeanWeighting
 
 
@@ -23,4 +23,4 @@
 )
 def test_pref_vector_to_weighting_check(pref_vector: Tensor | None, expectation: ExceptionContext):
     with expectation:
-        _ = _pref_vector_to_weighting(pref_vector, default=_MeanWeighting())
+        _ = pref_vector_to_weighting(pref_vector, default=_MeanWeighting())