Skip to content

Commit 8ec1f46

Browse files
committed
Fix deprecated yield warnings filter
1 parent 03bb303 commit 8ec1f46

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

graphdatascience/query_runner/neo4j_query_runner.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -329,7 +329,7 @@ def _handle_notifications(self, result_summary: neo4j.ResultSummary) -> None:
329329
status_objects = result_summary.gql_status_objects
330330
for status in status_objects:
331331
if status.raw_classification == "DEPRECATION" and not re.match(
332-
r".*returned by the procedure.*", status.status_description
332+
r"(.*returned by the procedure.*)|(.*procedure field deprecated.*)", status.status_description
333333
):
334334
warnings.warn(DeprecationWarning(status.status_description))
335335

@@ -458,7 +458,7 @@ def _verify_connectivity(
458458
else:
459459
warnings.filterwarnings(
460460
"ignore",
461-
category=neo4j.warnings.PreviewWarning,
461+
category=neo4j.warnings.PreviewWarning, # type: ignore[attr-defined]
462462
message=(r"^Passing key-word arguments to verify_connectivity\(\) is a preview feature.*"),
463463
)
464464

0 commit comments

Comments
 (0)