Skip to content

Commit 00d10ab

Browse files
committed
Internal change
PiperOrigin-RevId: 472663654
1 parent 22043a0 commit 00d10ab

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

shoptimizer_api/optimizers_builtin/condition_optimizer_test.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,7 @@
1515

1616
"""Unit tests for condition_optimizer.py."""
1717

18-
import unittest.mock as mock
18+
from unittest import mock
1919

2020
from absl.testing import parameterized
2121

@@ -29,6 +29,9 @@
2929
@mock.patch(
3030
'optimizers_builtin.condition_optimizer._GPC_STRING_TO_ID_MAPPING_CONFIG_FILE_NAME',
3131
'gpc_string_to_id_mapping_{}_test')
32+
@mock.patch(
33+
'optimizers_builtin.condition_optimizer._CONDITION_OPTIMIZER_CONFIG_FILENAME',
34+
'condition_optimizer_config_test')
3235
class ConditionOptimizerTest(parameterized.TestCase):
3336

3437
def setUp(self) -> None:

0 commit comments

Comments
 (0)