diff options
Diffstat (limited to 'src/pybind/mgr/localpool/module.py')
| -rw-r--r-- | src/pybind/mgr/localpool/module.py | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/src/pybind/mgr/localpool/module.py b/src/pybind/mgr/localpool/module.py index ce7cb1af2a6..907c7026b7d 100644 --- a/src/pybind/mgr/localpool/module.py +++ b/src/pybind/mgr/localpool/module.py @@ -1,4 +1,4 @@ -from mgr_module import MgrModule, CommandResult, Option +from mgr_module import MgrModule, CommandResult, Option, NotifyType import json import threading from typing import cast, Any @@ -51,8 +51,8 @@ class Module(MgrModule): super(Module, self).__init__(*args, **kwargs) self.serve_event = threading.Event() - def notify(self, notify_type: str, notify_id: str) -> None: - if notify_type == 'osd_map': + def notify(self, notify_type: NotifyType, notify_id: str) -> None: + if notify_type == NotifyType.osd_map: self.handle_osd_map() def handle_osd_map(self) -> None: |