feat: do not retry for client exceptions. code review

Signed-off-by: Ruben Romero Montes <rromerom@redhat.com>
NVIDIA-AI-Blueprints · Feb 3, 2025 · dfa0dd5 · dfa0dd5
1 parent 42f8d6c
commit dfa0dd5
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/cve/pipeline/input.py b/src/cve/pipeline/input.py
@@ -179,7 +179,7 @@ async def _inner():
 
             async with aiohttp.ClientSession() as session:
 
-                intel_retriever = IntelRetriever(session=session)
+                intel_retriever = IntelRetriever(session=session, retry_on_client_errors=run_config.general.retry_on_client_errors)
 
                 intel_coros = [intel_retriever.retrieve(vuln_id=cve.vuln_id) for cve in message.input.scan.vulns]
 
@@ -243,7 +243,7 @@ async def _inner():
 
             async with aiohttp.ClientSession() as session:
 
-                vuln_dep_checker = VulnerableDependencyChecker(session=session, image=image, sbom_list=sbom)
+                vuln_dep_checker = VulnerableDependencyChecker(session=session, image=image, sbom_list=sbom, retry_on_client_errors=run_config.general.retry_on_client_errors)
 
                 await vuln_dep_checker.load_dependencies()