From 00be35509add4388e7a8c8e8b8a28a94a6b30863 Mon Sep 17 00:00:00 2001
From: ncclementi <natyclementi@gmail.com>
Date: Wed, 31 Jul 2024 18:30:42 -0400
Subject: [PATCH] chore: fix druid exception in test_window

---
 ibis/backends/tests/test_window.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/ibis/backends/tests/test_window.py b/ibis/backends/tests/test_window.py
index 80756473f9ba..c1182a1d9bff 100644
--- a/ibis/backends/tests/test_window.py
+++ b/ibis/backends/tests/test_window.py
@@ -19,7 +19,6 @@
     MySQLOperationalError,
     PsycoPg2InternalError,
     Py4JJavaError,
-    PyDruidProgrammingError,
     PyODBCProgrammingError,
     SnowflakeProgrammingError,
 )
@@ -27,7 +26,7 @@
 
 pytestmark = [
     pytest.mark.notimpl(
-        ["druid"], raises=(com.OperationNotDefinedError, PyDruidProgrammingError)
+        ["druid"], raises=(com.OperationNotDefinedError, com.TableNotFound)
     )
 ]