Address PR comments

apache · Jan 25, 2021 · 526c5e4 · 526c5e4
1 parent 3c5ab2f
commit 526c5e4
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/superset/connectors/connector_registry.py b/superset/connectors/connector_registry.py
@@ -46,9 +46,9 @@ def register_sources(cls, datasource_config: "OrderedDict[str, List[str]]") -> N
     def get_datasource(
         cls, datasource_type: str, datasource_id: int, session: Session
     ) -> "BaseDatasource":
-        """Safely get a datasource without raising any errors.
-        Returns None if `datasource_type` is not registered or `datasource_id` does
-        not exists."""
+        """Safely get a datasource instance, raises `DatasetNotFoundError` if
+        `datasource_type` is not registered or `datasource_id` does not
+        exist."""
         if datasource_type not in cls.sources:
             raise DatasetNotFoundError()