From 31a63a3d57e4d2da0bfa2c1616d536625f2d252a Mon Sep 17 00:00:00 2001
From: Tim Swast <swast@google.com>
Date: Fri, 2 Aug 2024 16:07:22 +0000
Subject: [PATCH] feat: create a separate OrderingModePartialPreviewWarning for
 more fine-grained warning filters

---
 bigframes/exceptions.py       | 4 ++++
 bigframes/session/__init__.py | 2 +-
 2 files changed, 5 insertions(+), 1 deletion(-)

diff --git a/bigframes/exceptions.py b/bigframes/exceptions.py
index 1d31749760..ee597a3eaf 100644
--- a/bigframes/exceptions.py
+++ b/bigframes/exceptions.py
@@ -47,6 +47,10 @@ class NullIndexError(ValueError):
     """Object has no index."""
 
 
+class OrderingModePartialPreviewWarning(PreviewWarning):
+    """Ordering mode 'partial' is in preview."""
+
+
 class OrderRequiredError(ValueError):
     """Operation requires total row ordering to be enabled."""
 
diff --git a/bigframes/session/__init__.py b/bigframes/session/__init__.py
index 98cba867f2..b739efcc10 100644
--- a/bigframes/session/__init__.py
+++ b/bigframes/session/__init__.py
@@ -302,7 +302,7 @@ def __init__(
         if not self._strictly_ordered:
             warnings.warn(
                 "Partial ordering mode is a preview feature and is subject to change.",
-                bigframes.exceptions.PreviewWarning,
+                bigframes.exceptions.OrderingModePartialPreviewWarning,
             )
 
         # Sequential index needs total ordering to generate, so use null index with unstrict ordering.