typo

Lightning-AI · Jul 25, 2020 · 5aebae9 · 5aebae9
1 parent 3bb5414
commit 5aebae9
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tests/models/test_tpu.py b/tests/models/test_tpu.py
@@ -173,6 +173,7 @@ def test_model_16bit_tpu_index_5(tmpdir):
     model = EvalModelTemplate()
     trainer = Trainer(
         default_root_dir=tmpdir,
+        precision=16,
         max_epochs=1,
         train_percent_check=0.4,
         val_percent_check=0.2,