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.