diff --git a/metadata-ingestion/src/datahub/emitter/rest_emitter.py b/metadata-ingestion/src/datahub/emitter/rest_emitter.py index 2b5eea7308..fa7ea58c7d 100644 --- a/metadata-ingestion/src/datahub/emitter/rest_emitter.py +++ b/metadata-ingestion/src/datahub/emitter/rest_emitter.py @@ -61,13 +61,13 @@ class DataHubRestEmitter(Closeable): retry_max_times: Optional[int] = None, extra_headers: Optional[Dict[str, str]] = None, ca_certificate_path: Optional[str] = None, - server_telemetry_id: Optional[str] = None, disable_ssl_verification: bool = False, ): + if not gms_server: + raise ConfigurationError("gms server is required") self._gms_server = gms_server self._token = token self.server_config: Dict[str, Any] = {} - self.server_telemetry_id: str = "" self._session = requests.Session() diff --git a/metadata-ingestion/src/datahub_provider/hooks/datahub.py b/metadata-ingestion/src/datahub_provider/hooks/datahub.py index e8ae417ac3..e2e523fc5d 100644 --- a/metadata-ingestion/src/datahub_provider/hooks/datahub.py +++ b/metadata-ingestion/src/datahub_provider/hooks/datahub.py @@ -56,7 +56,7 @@ class DatahubRestHook(BaseHook): conn: "Connection" = self.get_connection(self.datahub_rest_conn_id) host = conn.host - if host is None: + if not host: raise AirflowException("host parameter is required") if conn.port: if ":" in host: