Skip to content

Commit 48f19d0

Browse files
authored
fix(attack-paths): neo4j.exceptions import (#9356)
1 parent 345033e commit 48f19d0

File tree

2 files changed

+2
-3
lines changed

2 files changed

+2
-3
lines changed

api/src/backend/api/attack_paths/database.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -6,11 +6,10 @@
66
from uuid import UUID
77

88
import neo4j
9+
import neo4j.exceptions
910

1011
from django.conf import settings
1112

12-
import neo4j.exceptions
13-
1413
from api.attack_paths.retryable_session import RetryableSession
1514

1615
# Without this Celery goes crazy with Neo4j logging

api/src/backend/api/attack_paths/retryable_session.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,5 @@
11
import logging
2+
23
from collections.abc import Callable
34
from typing import Any
45

@@ -18,7 +19,6 @@ def __init__(
1819
session_factory: Callable[[], neo4j.Session],
1920
close_driver: Callable[[], None], # Just to avoid circular imports
2021
max_retries: int,
21-
logger: logging.Logger | None = None,
2222
) -> None:
2323
self._session_factory = session_factory
2424
self._close_driver = close_driver

0 commit comments

Comments
 (0)