From 4da84776d8f6c3bf8967ffc73e2a7cf83286194a Mon Sep 17 00:00:00 2001 From: Furkan Date: Sat, 14 Dec 2024 22:52:25 +0300 Subject: [PATCH] fix: rename `DAQJobMessageStatsRemoteDict` to `DAQJobStatsRemoteDict` for consistency --- src/enrgdaq/daq/jobs/handle_stats.py | 4 ++-- src/enrgdaq/daq/jobs/remote.py | 6 +++--- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/src/enrgdaq/daq/jobs/handle_stats.py b/src/enrgdaq/daq/jobs/handle_stats.py index 306a746..dcc6ae2 100644 --- 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 index 4886d6f..1a896af 100644 --- 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)