diff --git a/socs/agents/lakeshore240/agent.py b/socs/agents/lakeshore240/agent.py index 1ae536fee..1fa7017db 100644 --- a/socs/agents/lakeshore240/agent.py +++ b/socs/agents/lakeshore240/agent.py @@ -6,10 +6,8 @@ import traceback import warnings from dataclasses import dataclass, fields -from typing import ( - Any, Dict, Generator, Optional, Tuple, Type, Callable, get_args, get_origin, - Union -) +from typing import (Any, Callable, Dict, Generator, Optional, Tuple, Type, + Union, get_args, get_origin) import txaio # type: ignore from ocs import ocs_agent, site_config @@ -157,6 +155,7 @@ def is_instanceable(t: Type) -> bool: except Exception: return False + def get_param_type(t: Type) -> Optional[Type]: """ Takes in a dataclass field type and returns a type that is accepted