fixing mypy issues

Author: radu-gheorghe

providers/vespa/src/airflow/providers/vespa/hooks/vespa.py

@@ -16,7 +16,6 @@
 # under the License.
 from __future__ import annotations
 
-import types
 import uuid
 from collections.abc import Callable, Iterable
 from queue import Empty, Queue
@@ -94,12 +93,16 @@ def __init__(
         conn: Connection = self.get_connection(conn_id)
         self.conn = conn
         extra = self.conn.extra_dejson or {}
+        host = self.conn.host
+
+        if not host:
+            raise ValueError("Vespa connection requires a host")
 
         resolved_schema = schema or self.conn.schema or self._get_field(extra, "schema")
         resolved_namespace = namespace or self._get_field(extra, "namespace") or "default"
 
         self._configure_from_connection(
-            host=self.conn.host,
+            host=host,
             port=self.conn.port,
             namespace=resolved_namespace,
             schema=resolved_schema,
@@ -126,8 +129,6 @@ def from_resolved_connection(
         self = cls.__new__(cls)
         BaseHook.__init__(self)
 
-        self.conn = types.SimpleNamespace(host=host, extra=extra)
-
         resolved_schema = schema or cls._get_field(extra, "schema")
         resolved_namespace = namespace or cls._get_field(extra, "namespace") or "default"
 

providers/vespa/tests/system/vespa/example_dag_vespa.py

@@ -104,9 +104,9 @@ def run_query(yql: str, params: dict | None = None):
     send_batches >> [update_doc3, delete_doc4] >> verify_docs_task
 
 
-dag = vespa_dynamic()
+example_dag = vespa_dynamic()
 
 
 from tests_common.test_utils.system_tests import get_test_run  # noqa: E402
 
-test_run = get_test_run(dag)
+test_run = get_test_run(example_dag)