Fix bugs

Author: Liu Keyu

noxfile.py

@@ -67,7 +67,9 @@ def _run_tests(
         "test",
         *install_args,
         "pytest",
+        "-v",
         *pytest_run_args,
+        "--disable-warnings",
         *session.posargs,
         "--cov-config=pyproject.toml",
         env=env,
@@ -137,5 +139,7 @@ def docs(session: nox.Session) -> None:
         "--frozen",
         "sphinx-autobuild" if serve else "sphinx-build",
         *shared_args,
+        "-v",
+        "--disable-warnings",
         env=env,
     )

pyproject.toml

@@ -44,7 +44,7 @@ dependencies = [
     "numpy>=1.24; python_version >= '3.11'",
     "numpy>=1.22",
     "numpy>=1.22,<2; sys_platform == 'darwin' and 'x86_64' in platform_machine and python_version < '3.13'",  # Restrict numpy v2 for macOS x86 since it is not supported anymore since torch v2.3.0
-    "torch>=2.7.1,<2.8.0; sys_platform == 'darwin' and 'x86_64' in platform_machine and python_version < '3.13'",  # Restrict torch v2.3.0 for macOS x86 since it is not supported anymore.
+    "torch>=2.7.1,<2.8.0",
     "typing-extensions>=4.1",  # for `assert_never`
     "qiskit-ibm-transpiler>=0.11.1",
     "qiskit-ibm-ai-local-transpiler>=0.3.2; python_version < '3.13'"

src/mqt/predictor/ml/predictor.py

@@ -205,8 +205,7 @@ def compile_training_circuits(
         self,
         path_uncompiled_circuits: Path | None = None,
         path_compiled_circuits: Path | None = None,
-        timeout: int = 600,
-        num_workers: int = -1,
+        timeout: int = 6000,
     ) -> None:
         """Compiles all circuits in the given directory with the given timeout and saves them in the given directory.
 
@@ -227,7 +226,7 @@ def compile_training_circuits(
             with zipfile.ZipFile(str(path_zip), "r") as zip_ref:
                 zip_ref.extractall(path_uncompiled_circuits)
 
-        Parallel(n_jobs=num_workers, verbose=100)(
+        Parallel(n_jobs=1, verbose=100)(
             delayed(self._compile_all_circuits_devicewise)(
                 device, timeout, path_uncompiled_circuits, path_compiled_circuits, logger.level
             )
@@ -239,7 +238,6 @@ def generate_training_data(
         path_uncompiled_circuits: Path | None = None,
         path_compiled_circuits: Path | None = None,
         path_training_data: Path | None = None,
-        num_workers: int = -1,
     ) -> None:
         """Creates and saves training data from all generated training samples.
 
@@ -267,7 +265,7 @@ def generate_training_data(
         names_list = []
         scores_list = []
 
-        results = Parallel(n_jobs=num_workers, verbose=100)(
+        results = Parallel(n_jobs=1, verbose=100)(
             delayed(self._generate_training_sample)(
                 filename.name,
                 path_uncompiled_circuits,

src/mqt/predictor/rl/actions.py

@@ -495,6 +495,7 @@ def get_openqasm_gates() -> list[str]:
         PassType.MAPPING,
         stochastic=True,
         transpile_pass=lambda device: [
+            ### Requires a initial layout, but "optimize" mode overwrites it
             TrivialLayout(coupling_map=CouplingMap(device.build_coupling_map())),
             FullAncillaAllocation(coupling_map=CouplingMap(device.build_coupling_map())),
             EnlargeWithAncilla(),

src/mqt/predictor/rl/predictorenv.py

@@ -341,7 +341,6 @@ def _apply_qiskit_action(self, action: Action, action_index: int) -> QuantumCirc
             def metric_fn(circ: QuantumCircuit) -> float:
                 return float(circ.count_ops().get("swap", 0))
 
-            # for stochastic actions, pass the layout/routing trials parameter
             max_iteration = self.max_iter
             if "Sabre" in action.name and "AIRouting" not in action.name:
                 # Internal trials for Sabre
@@ -359,6 +358,10 @@ def metric_fn(circ: QuantumCircuit) -> float:
                     max_iteration=max_iteration,
                     metric_fn=metric_fn,
                 )
+            else:
+                msg = f"Unknown stochastic action: {action.name}"
+                raise ValueError(msg)
+
         else:
             if action.name in ["QiskitO3", "Opt2qBlocks"] and isinstance(action, DeviceDependentAction):
                 passes = action.transpile_pass(
@@ -378,8 +381,8 @@ def metric_fn(circ: QuantumCircuit) -> float:
                 pm = PassManager(transpile_pass)
                 altered_qc = pm.run(self.state)
                 pm_property_set = dict(pm.property_set) if hasattr(pm, "property_set") else {}
+
         if action_index in (self.actions_mapping_indices + self.actions_final_optimization_indices):
-            pm_property_set = dict(pm.property_set)
             altered_qc = self._handle_qiskit_layout_postprocessing(action, pm_property_set, altered_qc)
 
         return altered_qc