From 0bca178412ffafd14780f0f9d49b39f3b89a4cd3 Mon Sep 17 00:00:00 2001 From: Lukas Prediger Date: Thu, 12 Jul 2018 14:05:25 +0200 Subject: [PATCH] Generalizing type annotations for set/get_default_pulse_registry() and Serializable.__init__() Signed-off-by: Lukas Prediger --- qctoolkit/serialization.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/qctoolkit/serialization.py b/qctoolkit/serialization.py index d9cc30545..6580e867c 100644 --- a/qctoolkit/serialization.py +++ b/qctoolkit/serialization.py @@ -10,7 +10,7 @@ """ from abc import ABCMeta, abstractmethod -from typing import Dict, Any, Optional, NamedTuple, Union +from typing import Dict, Any, Optional, NamedTuple, Union, MutableMapping import os import zipfile import tempfile @@ -321,11 +321,11 @@ def __new__(mcs, name, bases, dct): default_pulse_registry = None -def get_default_pulse_registry() -> Union[Dict, 'PulseStorage']: +def get_default_pulse_registry() -> MutableMapping: return default_pulse_registry -def set_default_pulse_registry(new_default_registry: Optional[Union[Dict, 'PulseStorage']]) -> None: +def set_default_pulse_registry(new_default_registry: Optional[MutableMapping]) -> None: global default_pulse_registry default_pulse_registry = new_default_registry @@ -358,7 +358,7 @@ class Serializable(metaclass=SerializableMeta): type_identifier_name = '#type' identifier_name = '#identifier' - def __init__(self, identifier: Optional[str]=None, registry: Optional[dict]=None) -> None: + def __init__(self, identifier: Optional[str]=None, registry: Optional[MutableMapping]=None) -> None: """Initialize a Serializable. Args: