Fix a few flake8 issues

mfeurer · mfeurer · commit a6d6cfe22478 · 2023-04-18T12:55:32.000+02:00
diff --git a/openml/extensions/extension_interface.py b/openml/extensions/extension_interface.py
@@ -166,7 +166,7 @@ def _run_model_on_fold(
         y_train: Optional[np.ndarray] = None,
         X_test: Optional[Union[np.ndarray, scipy.sparse.spmatrix]] = None,
     ) -> Tuple[np.ndarray, np.ndarray, "OrderedDict[str, float]", Optional["OpenMLRunTrace"]]:
-        """Run a model on a repeat,fold,subsample triplet of the task and return prediction information.
+        """Run a model on a repeat, fold, subsample triplet of the task.
 
         Returns the data that is necessary to construct the OpenML Run object. Is used by
         :func:`openml.runs.run_flow_on_task`.
diff --git a/openml/study/study.py b/openml/study/study.py
@@ -99,11 +99,11 @@ def id(self) -> Optional[int]:
 
     def _get_repr_body_fields(self) -> List[Tuple[str, Union[str, int, List[str]]]]:
         """Collect all information to display in the __repr__ body."""
-        fields = {
+        fields: Dict[str, Any] = {
             "Name": self.name,
             "Status": self.status,
             "Main Entity Type": self.main_entity_type,
-        }  # type: Dict[str, Any]
+        }
         if self.study_id is not None:
             fields["ID"] = self.study_id
             fields["Study URL"] = self.openml_url
diff --git a/openml/tasks/task.py b/openml/tasks/task.py
@@ -82,11 +82,11 @@ def id(self) -> Optional[int]:
 
     def _get_repr_body_fields(self) -> List[Tuple[str, Union[str, int, List[str]]]]:
         """Collect all information to display in the __repr__ body."""
-        fields = {
+        fields: Dict[str, Any] = {
             "Task Type Description": "{}/tt/{}".format(
                 openml.config.get_server_base_url(), self.task_type_id
             )
-        }  # type: Dict[str, Any]
+        }
         if self.task_id is not None:
             fields["Task ID"] = self.task_id
             fields["Task URL"] = self.openml_url
diff --git a/openml/utils.py b/openml/utils.py
@@ -72,13 +72,13 @@ def extract_xml_tags(xml_tag_name, node, allow_none=True):
 
 def _get_rest_api_type_alias(oml_object: "OpenMLBase") -> str:
     """Return the alias of the openml entity as it is defined for the REST API."""
-    rest_api_mapping = [
+    rest_api_mapping: List[Tuple[Union[Type, Tuple], str]] = [
         (openml.datasets.OpenMLDataset, "data"),
         (openml.flows.OpenMLFlow, "flow"),
         (openml.tasks.OpenMLTask, "task"),
         (openml.runs.OpenMLRun, "run"),
         ((openml.study.OpenMLStudy, openml.study.OpenMLBenchmarkSuite), "study"),
-    ]  # type: List[Tuple[Union[Type, Tuple], str]]
+    ]
     _, api_type_alias = [
         (python_type, api_alias)
         for (python_type, api_alias) in rest_api_mapping
