Skip to content

Commit 68b638f

Browse files
AleksMattensorflower-gardener
authored andcommitted
Automated Code Change
PiperOrigin-RevId: 937681848
1 parent 6b15c7b commit 68b638f

1 file changed

Lines changed: 8 additions & 9 deletions

File tree

tf_keras/distribute/mirrored_variable_test.py

Lines changed: 8 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -47,15 +47,14 @@ def get_strategy_with_mimicing_cpus():
4747

4848
@tf.__internal__.distribute.combinations.generate(
4949
tf.__internal__.test.combinations.combine(
50-
distribution=list(
51-
filter(
52-
None.__ne__,
53-
[
54-
tf.__internal__.distribute.combinations.mirrored_strategy_with_gpu_and_cpu, # noqa: E501
55-
get_strategy_with_mimicing_cpus(),
56-
],
57-
)
58-
),
50+
distribution=[
51+
x
52+
for x in [
53+
tf.__internal__.distribute.combinations.mirrored_strategy_with_gpu_and_cpu, # noqa: E501
54+
get_strategy_with_mimicing_cpus(),
55+
]
56+
if x is not None
57+
],
5958
mode=["graph", "eager"],
6059
)
6160
)

0 commit comments

Comments
 (0)