fix: rename DAQJobMessageStatsRemoteDict to DAQJobStatsRemoteDict…

… for consistency
ENRG-tr · Dec 14, 2024 · 4da8477 · 4da8477
1 parent b9b2e88
commit 4da8477
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/src/enrgdaq/daq/jobs/handle_stats.py b/src/enrgdaq/daq/jobs/handle_stats.py
@@ -7,7 +7,7 @@
 from enrgdaq.daq.base import DAQJob
 from enrgdaq.daq.jobs.remote import (
     DAQJobMessageStatsRemote,
-    DAQJobMessageStatsRemoteDict,
+    DAQJobStatsRemoteDict,
     SupervisorRemoteStats,
 )
 from enrgdaq.daq.models import DAQJobMessage, DAQJobStats, DAQJobStatsRecord
@@ -48,7 +48,7 @@ class DAQJobHandleStats(DAQJob):
     config: DAQJobHandleStatsConfig
 
     _stats: dict[str, DAQJobStatsDict]
-    _remote_stats: dict[str, DAQJobMessageStatsRemoteDict]
+    _remote_stats: dict[str, DAQJobStatsRemoteDict]
 
     def __init__(self, config: DAQJobHandleStatsConfig, **kwargs):
         super().__init__(config, **kwargs)

diff --git a/src/enrgdaq/daq/jobs/remote.py b/src/enrgdaq/daq/jobs/remote.py
@@ -46,10 +46,10 @@ def update_message_out_stats(self, message_out_bytes: int):
 class DAQJobMessageStatsRemote(DAQJobMessage):
     """Message class containing remote statistics."""
 
-    stats: "DAQJobMessageStatsRemoteDict"
+    stats: "DAQJobStatsRemoteDict"
 
 
-DAQJobMessageStatsRemoteDict = dict[str, SupervisorRemoteStats]
+DAQJobStatsRemoteDict = dict[str, SupervisorRemoteStats]
 
 
 class DAQJobRemoteConfig(DAQJobConfig):
@@ -94,7 +94,7 @@ class DAQJobRemote(DAQJob):
     _message_class_cache: dict[str, type[DAQJobMessage]]
     _remote_message_ids: set[str]
     _receive_thread: threading.Thread
-    _remote_stats: DAQJobMessageStatsRemoteDict
+    _remote_stats: DAQJobStatsRemoteDict
 
     def __init__(self, config: DAQJobRemoteConfig, **kwargs):
         super().__init__(config, **kwargs)