type hint fixes for adaptive/learner/data_saver.py

Author: basnijholt

adaptive/learner/data_saver.py

@@ -1,7 +1,7 @@
 import functools
 from collections import OrderedDict
 from operator import itemgetter
-from typing import Callable, Dict, Tuple, Union
+from typing import Any, Dict, Tuple, Union
 
 from adaptive.learner.base_learner import BaseLearner
 from adaptive.utils import copy_docstring_from
@@ -27,13 +27,13 @@ class DataSaver:
     >>> learner = DataSaver(_learner, arg_picker=itemgetter('y'))
     """
 
-    def __init__(self, learner: BaseLearner, arg_picker: itemgetter,) -> None:
+    def __init__(self, learner: BaseLearner, arg_picker: itemgetter) -> None:
         self.learner = learner
         self.extra_data = OrderedDict()
         self.function = learner.function
         self.arg_picker = arg_picker
 
-    def __getattr__(self, attr: str) -> Union[Callable, int]:
+    def __getattr__(self, attr: str) -> Any:
         return getattr(self.learner, attr)
 
     @copy_docstring_from(BaseLearner.tell)
@@ -46,13 +46,7 @@ def tell(self, x, result):
     def tell_pending(self, x):
         self.learner.tell_pending(x)
 
-    def _get_data(
-        self,
-    ) -> Union[
-        Tuple[Dict[Union[int, float], float], OrderedDict],
-        Tuple[OrderedDict, OrderedDict],
-        Tuple[Tuple[Dict[int, float], int, float, float], OrderedDict],
-    ]:
+    def _get_data(self,) -> Tuple[Any, OrderedDict]:
         return self.learner._get_data(), self.extra_data
 
     def _set_data(