diff --git a/tensorflow/python/distribute/mirrored_strategy.py b/tensorflow/python/distribute/mirrored_strategy.py index d52eae917ba..2fefdeac6d7 100644 --- a/tensorflow/python/distribute/mirrored_strategy.py +++ b/tensorflow/python/distribute/mirrored_strategy.py @@ -355,8 +355,8 @@ class MirroredExtended(distribute_lib.StrategyExtendedV1): self._initialize_single_worker(devices) if self._prefer_collective_ops and ( isinstance(self._cross_device_ops, cross_device_ops_lib.NcclAllReduce) - or isinstance(self._inferred_cross_device_ops), - cross_device_ops_lib.NcclAllReduce): + or isinstance(self._inferred_cross_device_ops, + cross_device_ops_lib.NcclAllReduce)): self._use_collective_ops(devices) self._inferred_cross_device_ops = None logging.info("Using MirroredStrategy with devices %r", devices)