From 10455f4ee2db1edcd47c6005a806afe6abb4be70 Mon Sep 17 00:00:00 2001 From: Grain Team Date: Fri, 30 Jan 2026 07:32:06 -0800 Subject: [PATCH] Explicitly set `py_traceback_filtering` to "auto" in relevant tests. PiperOrigin-RevId: 863231903 --- grain/_src/core/traceback_util_test.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/grain/_src/core/traceback_util_test.py b/grain/_src/core/traceback_util_test.py index 1ac0e6093..25237425b 100644 --- a/grain/_src/core/traceback_util_test.py +++ b/grain/_src/core/traceback_util_test.py @@ -68,6 +68,7 @@ def raise_after_recursion(curr_count: int, max_count: int) -> None: class TracebackUtilTest(absltest.TestCase): def test_traceback_mode_auto_decorator_filters_traceback(self): + config.update("py_traceback_filtering", "auto") _assert_exception_with_short_traceback( self, lambda: raise_after_recursion(0, 150), ValueError ) @@ -160,6 +161,7 @@ def map(self, x: int) -> int: class TracebackFilterTest(absltest.TestCase): def test_datasource_multiple_transforms_filters_traceback(self): + config.update("py_traceback_filtering", "auto") range_ds = grain.sources.RangeDataSource(0, 10, 1) sampler = grain.samplers.IndexSampler(num_records=10, seed=42) ops = [RaiseErrorTransform()] @@ -173,6 +175,7 @@ def test_datasource_multiple_transforms_filters_traceback(self): ) def test_dataset_multiple_transforms_filters_traceback(self): + config.update("py_traceback_filtering", "auto") range_ds = grain.MapDataset.range(0, 10) range_ds = range_ds.map(RaiseErrorTransform()) for _ in range(100):