fix(attack-paths): neo4j.exceptions import (#9356)

This commit is contained in:
Josema Camacho
2025-12-01 10:31:18 +01:00
committed by GitHub
parent 345033e58a
commit 48f19d0f11
2 changed files with 2 additions and 3 deletions
+1 -2
View File
@@ -6,11 +6,10 @@ from typing import Iterator
from uuid import UUID
import neo4j
import neo4j.exceptions
from django.conf import settings
import neo4j.exceptions
from api.attack_paths.retryable_session import RetryableSession
# Without this Celery goes crazy with Neo4j logging
@@ -1,4 +1,5 @@
import logging
from collections.abc import Callable
from typing import Any
@@ -18,7 +19,6 @@ class RetryableSession:
session_factory: Callable[[], neo4j.Session],
close_driver: Callable[[], None], # Just to avoid circular imports
max_retries: int,
logger: logging.Logger | None = None,
) -> None:
self._session_factory = session_factory
self._close_driver = close_driver