Skip to content
This repository was archived by the owner on Jul 1, 2023. It is now read-only.

Commit 599768e

Browse files
committed
use assertEqual instead of XCT assert true in huber loss test case
1 parent fe4db75 commit 599768e

File tree

1 file changed

+3
-5
lines changed

1 file changed

+3
-5
lines changed

Tests/TensorFlowTests/LossTests.swift

Lines changed: 3 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -225,13 +225,11 @@ final class LossTests: XCTestCase {
225225

226226
// test huber(x, x) = 0
227227
let loss1 = huberLoss(predicted: predictions, expected: predictions, delta: Float(1))
228-
XCTAssertTrue(loss1.isAlmostEqual(to: Tensor(0), tolerance: 0.001), "Huber Loss all correct")
228+
assertEqual(loss1, Tensor(0), accuracy: 1e-6)
229229

230-
// test huber(p, t) = 0.62500006 computed from tf.keras.losses.Huber
230+
// test huber(p, t) = 0.62500006 computed from tf.keras.losses.Huber(delta=1)
231231
let loss2 = huberLoss(predicted: predictions, expected: targets, delta: Float(1))
232-
XCTAssertTrue(
233-
loss2.isAlmostEqual(to: Tensor(0.62500006), tolerance: 0.001),
234-
"Huber Loss unweighted")
232+
assertEqual(loss2, Tensor(0.62500006), accuracy: 1e-6)
235233
}
236234

237235
static var allTests = [

0 commit comments

Comments
 (0)