de-duplicate _infer_outputs logic

Author: RoyStegeman

src/qibolab/_core/instruments/qblox/platform.py

@@ -111,38 +111,27 @@ def _out_port(port: Union[str, int]) -> QubitId:
     return port if isinstance(port, int) else _digits(port)
 
 
-def infer_los(cluster: dict) -> dict[tuple[QubitId, bool], str]:
-    """Infer LOs names for output channels.
-
-    ``cluster`` should be a mapping compatible with the same input of :func:`map_ports`.
+def _infer_outputs(cluster: dict, suffix: str) -> dict[tuple[QubitId, bool], str]:
+    """``cluster`` should be a mapping compatible with the same input of :func:`map_ports`.
 
     The result is a mapping from ``(qubit, channel)``, where ``qubit`` is the identifier,
     and ``channel`` is a boolean toggle: ``True`` for probe channels, ``False`` for
     drive.
     """
     return {
-        (q, "qrm" in mod): f"{mod}/o{_out_port(port)}/lo"
+        (q, "qrm" in mod): f"{mod}/o{_out_port(port)}/{suffix}"
         for mod, specs in cluster.items()
         if "_rf" in mod
         for port, qs in specs[1].items()
         for q in qs
     }
 
 
-def infer_mixers(cluster: dict) -> dict[tuple[QubitId, bool], str]:
-    """Infer mixer names for output channels.
+def infer_los(cluster: dict) -> dict[tuple[QubitId, bool], str]:
+    """Infer LOs names for output channels."""
+    return _infer_outputs(cluster, "lo")
 
-    ``cluster`` should be a mapping compatible with the same input of :func:`map_ports`.
 
-    The result is a mapping from ``(qubit, channel)``, where ``qubit`` is the identifier,
-    and ``channel`` is a boolean toggle: ``True`` for probe channels, ``False`` for
-    drive.
-    """
-    # TODO: same as infer_los, avoid duplicating
-    return {
-        (q, "qrm" in mod): f"{mod}/o{_out_port(port)}/mixer"
-        for mod, specs in cluster.items()
-        if "_rf" in mod
-        for port, qs in specs[1].items()
-        for q in qs
-    }
+def infer_mixers(cluster: dict) -> dict[tuple[QubitId, bool], str]:
+    """Infer mixer names for output channels."""
+    return _infer_outputs(cluster, "mixer")