not defined variables

patrick-toulme · Dec 11, 2024 · 1bdb44e · 1bdb44e
1 parent ee89ed3
commit 1bdb44e
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/axlearn/common/inference_test.py b/axlearn/common/inference_test.py
@@ -173,7 +173,6 @@ def is_supported(
     global_batch_size: int,
     data_partition: DataPartitionType,
 ):
-    del param_dtype, use_ema  # not used
     # TODO(xuan-zou): jax 0.4.25 breaks bfloat16 on CPU due to high variance on
     # the final result (up to 10% precision diff), will re-enable when fixed.
     # NOTE: bfloat16 test on GPU is added and verified.