fixed bug in type declaration

cdplib/hyperopt/HyperoptPipelineSelector.py

@@ -184,7 +184,7 @@ class HyperoptPipelineSelector(PipelineSelector):
             self._logger.log_and_raise_error(err)
 
     def _get_space_element_from_trial(self, trial: dict)\
-            -> Union[Dict[SpaceElementType], None]:
+            -> Union[Dict[str, SpaceElementType], None]:
         """
         Hyperopt trials object does not contain the space
              elements that result in the corresponding trials.
@@ -220,7 +220,7 @@ class HyperoptPipelineSelector(PipelineSelector):
             self._logger.log_and_raise_error(err)
 
     def _get_space_element_from_index(self, i: int)\
-            -> Union[Dict[SpaceElementType], None]:
+            -> Union[Dict[str, SpaceElementType], None]:
         """
         Gets the space element of shape
         {"name": NAME, "params": PARAMS, "pipeline": PIPELINE}