add a type for pycharms linter

Akshita07 · Jun 10, 2020 · 5d7cd07 · 5d7cd07
1 parent caf3947
commit 5d7cd07
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/qcodes/tests/parameter/test_snapshot.py b/qcodes/tests/parameter/test_snapshot.py
@@ -1,4 +1,4 @@
-from typing import Union, Optional
+from typing import Union, Optional, Callable
 from datetime import datetime, timedelta
 
 from qcodes.instrument.parameter import Parameter
@@ -8,7 +8,7 @@
 def create_parameter(snapshot_get: bool,
                      snapshot_value: bool,
                      cache_is_valid: bool,
-                     get_cmd: Optional[bool],
+                     get_cmd: Optional[Union[Callable, bool]],
                      offset: Union[str, float] = NOT_PASSED):
     kwargs = dict(set_cmd=None,
                   label='Parameter',