Skip to content

Commit 6bb3e8d

Browse files
committed
Remove import not existing in neo4j 4.4
1 parent 23eb310 commit 6bb3e8d

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

graphdatascience/tests/integration/test_error_handling.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
from typing import Generator
22

33
import pytest
4-
from neo4j import Driver, GraphDatabase, NotificationMinimumSeverity
4+
from neo4j import Driver, GraphDatabase
55

66
from graphdatascience import GraphDataScience
77
from graphdatascience.query_runner.neo4j_query_runner import Neo4jQueryRunner
@@ -35,7 +35,7 @@ def run_around_tests(gds: GraphDataScience) -> Generator[None, None, None]:
3535

3636
@pytest.fixture
3737
def warning_driver() -> Generator[Driver, None, None]:
38-
driver = GraphDatabase.driver(URI, auth=AUTH, warn_notification_severity=NotificationMinimumSeverity.WARNING)
38+
driver = GraphDatabase.driver(URI, auth=AUTH, warn_notification_severity="WARNING")
3939

4040
yield driver
4141

0 commit comments

Comments
 (0)