Rename gather_mcm_jit

PennyLaneAI · Jun 3, 2024 · ef4fc27 · ef4fc27
1 parent b5c1415
commit ef4fc27
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pennylane/transforms/dynamic_one_shot.py b/pennylane/transforms/dynamic_one_shot.py
@@ -277,7 +277,7 @@ def measurement_with_no_shots(measurement):
         if interface != "jax" and m.mv and not has_valid:
             meas = measurement_with_no_shots(m)
         elif m.mv and active_qjit:
-            meas = gather_mcm_jit(m, mcm_samples, is_valid)
+            meas = gather_mcm_qjit(m, mcm_samples, is_valid)
         elif m.mv:
             meas = gather_mcm(m, mcm_samples, is_valid)
         elif interface != "jax" and not has_valid:
@@ -307,7 +307,7 @@ def measurement_with_no_shots(measurement):
     return tuple(normalized_meas) if len(normalized_meas) > 1 else normalized_meas[0]
 
 
-def gather_mcm_jit(measurement, samples, is_valid):
+def gather_mcm_qjit(measurement, samples, is_valid):
     """Process MCM measurements when the Catalyst compiler is active.
 
     Args: