Import the SpawnProcess class from multiprocessing.context

ml-energy · Dec 9, 2024 · 8c5456e · 8c5456e
1 parent f8d5b67
commit 8c5456e
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/zeus/metric.py b/zeus/metric.py
@@ -7,6 +7,7 @@
 import warnings
 from typing import Sequence
 import multiprocessing as mp
+from multiprocessing.context import SpawnProcess
 from dataclasses import dataclass
 
 from prometheus_client import (
@@ -28,7 +29,7 @@ class MonitoringProcessState:
     """Represents the state of a monitoring window."""
 
     queue: mp.Queue
-    proc: mp.context.SpawnProcess
+    proc: SpawnProcess
 
 
 class Metric(abc.ABC):