type drafts

nrm/coolr/clr_rapl.pyi

@@ -7,3 +7,9 @@ class rapl_reader:
 
     def read_config(self) -> RAPLConfig:
         ...
+
+    def setpcaps(self, control: PcapControl) -> None:
+        ...
+
+    def setfrequencies(self, control: FreqControl) -> None:
+        ...

nrm/nodemanager.py

@@ -26,7 +26,7 @@ import nrm.coolr.clr_misc
 
 from typing import Any, Dict, NamedTuple
 
-from nrm.types import MachineInfo
+from nrm.types import MachineInfo, FreqControl, PcapControl
 
 
 class NodeManager:
@@ -47,16 +47,14 @@ class NodeManager:
         return MachineInfo(
             energySamples=self._rapl.sample(),
             tempSamples=self._coretemp.sample(),
-            time=time.time()
+            time=time.time(),
         )
 
     def read_config(self):
         self._rapl.read_config()
 
-    def do_control_pcap(self, PcapControl) -> None:
-        self._rapl.setpcaps()
-        pass
+    def do_control_pcap(self, control: PcapControl) -> None:
+        self._rapl.setpcaps(control)
 
     def do_control_ddcm(self, control: FreqControl) -> None:
-        self._rapl.setfrequencies()
-        pass
+        self._rapl.setfrequencies(control)