test: fix ci test

datahub-project · Oct 21, 2024 · 9835e31 · 9835e31
1 parent 6574b63
commit 9835e31
Show file tree

Hide file tree

Showing 3 changed files with 945 additions and 978 deletions.
diff --git a/metadata-ingestion/src/datahub/ingestion/source/dremio/dremio_api.py b/metadata-ingestion/src/datahub/ingestion/source/dremio/dremio_api.py
@@ -87,7 +87,7 @@ def _setup_session(self) -> None:
         """Setup the session for retries and connection handling."""
         retry_strategy = Retry(
             total=self._retry_count,
-            status_forcelist=[429, 500, 502, 503, 504],
+            status_forcelist=[429, 502, 503, 504],
             method_whitelist=["HEAD", "GET", "OPTIONS", "POST"],
             backoff_factor=1,
         )
@@ -185,7 +185,6 @@ def post(self, url: str, data: str) -> Dict:
             verify=self._verify,
             timeout=self._timeout,
         )
-        print("url", self.base_url + url, self.session.headers, response.text)
         return response.json()
 
     def execute_query(self, query: str, timeout: int = 300) -> List[Dict[str, Any]]:

diff --git a/metadata-ingestion/src/datahub/ingestion/source/dremio/dremio_entities.py b/metadata-ingestion/src/datahub/ingestion/source/dremio/dremio_entities.py
@@ -72,6 +72,7 @@
     "DATA_MANIPULATION": ["INSERT INTO", "MERGE INTO", "CREATE TABLE"],
 }
 
+
 @dataclass
 class DremioContainerResponse:
     container_type: str