Update to new PMT Python interface

MiloLurati · Jun 5, 2024 · e21681a · e21681a
1 parent c24217a
commit e21681a
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/kernel_tuner/observers/pmt.py b/kernel_tuner/observers/pmt.py
@@ -48,7 +48,7 @@ def __init__(self, observable=None):
             if not obs in supported:
                 raise ValueError(f"Observable {obs} not in supported: {supported}")
 
-        self.pms = [pmt.get_pmt(obs[0], obs[1]) for obs in observable.items()]
+        self.pms = [pmt.create(obs[0], obs[1]) for obs in observable.items()]
         self.pm_names = list(observable.keys())
 
         self.begin_states = [None] * len(self.pms)
@@ -70,9 +70,8 @@ def after_finish(self):
         for i in range(len(self.pms)):
             begin_state = self.begin_states[i]
             end_state = end_states[i]
-            measured_energy = pmt.pypmt.PMT.joules(begin_state, end_state)
-            measured_power = pmt.pypmt.PMT.watts(begin_state, end_state)
-            print(measured_energy, measured_power)
+            measured_energy = pmt.joules(begin_state, end_state)
+            measured_power = pmt.watts(begin_state, end_state)
             pm_name = self.pm_names[i]
             energy_result_name = f"{pm_name}_energy"
             power_result_name = f"{pm_name}_power"