Amend the naming of the function.

huggingface · Jun 5, 2024 · b26d202 · b26d202
1 parent f888ce3
commit b26d202
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/test_object_detection.py b/tests/test_object_detection.py
@@ -64,7 +64,7 @@ def test_inference_default(self):
  expected_location = np.array([344.0622, 24.8543, 640.3398, 373.7401])
  self.assertLess(np.abs(boxes[0].cpu().detach().numpy() - expected_location).max(), 1)
 
- def test_inference_bf16(self):
+ def test_inference_autocast(self):
  model, processor = self.prepare_model_and_processor()
  image = self.prepare_data()
  inputs = processor(images=image, return_tensors="pt").to("hpu")
@@ -93,7 +93,7 @@ def test_inference_hpu_graphs(self):
  expected_location = np.array([344.0622, 24.8543, 640.3398, 373.7401])
  self.assertLess(np.abs(boxes[0].to(torch.float32).cpu().detach().numpy() - expected_location).max(), 1)
 
- def test_no_latency_regression_bf16(self):
+ def test_no_latency_regression_autocast(self):
  warmup = 3
  iterations = 10