Fix missing tpu variable in benchmark_args_tf.py (#13968)

85d69a7d · Hardian Lawi · GitHub · 990de2c1 · 85d69a7d
Unverified Commit 85d69a7d authored Oct 12, 2021 by Hardian Lawi Committed by GitHub Oct 11, 2021
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 0 deletions

src/transformers/benchmark/benchmark_args_tf.py src/transformers/benchmark/benchmark_args_tf.py +1 -0

No files found.
--- a/src/transformers/benchmark/benchmark_args_tf.py
+++ b/src/transformers/benchmark/benchmark_args_tf.py
@@ -79,6 +79,7 @@ class TensorFlowBenchmarkArguments(BenchmarkArguments):
    @cached_property
    @tf_required
    def _setup_tpu(self) -> Tuple["tf.distribute.cluster_resolver.TPUClusterResolver"]:
+        tpu = None
        if self.tpu:
            try:
                if self.tpu_name: