fix mypy

Bears-R-Us · Nov 20, 2024 · a2ef2d3 · a2ef2d3
1 parent 104eeb1
commit a2ef2d3
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/arkouda/client.py b/arkouda/client.py
@@ -1065,7 +1065,7 @@ def get_max_array_rank() -> int:
     return int(serverConfig["maxArrayDims"])
 
 
-def get_array_ranks() -> int:
+def get_array_ranks() -> list[int]:
     """
     Get the list of pdarray ranks the server was compiled to support
 
@@ -1108,7 +1108,7 @@ def _get_config_msg() -> Mapping[str, Union[str, int, float]]:
         raise RuntimeError(f"{e} in retrieving Arkouda server config")
 
 
-def _get_registration_config_msg() -> Mapping[str, Union[str, int, float, list]]:
+def _get_registration_config_msg() -> dict:
     """
     Get runtime information about the command registration configuration.