🎨 Fix formatting

Signed-off-by: gkumbhat <kumbhat.gaurav@gmail.com>
caikit · Jul 17, 2024 · b4e5108 · b4e5108
1 parent 9e3dda6
commit b4e5108
Showing 1 changed file with 2 additions and 6 deletions.
diff --git a/tests/modules/text_embedding/test_embedding.py b/tests/modules/text_embedding/test_embedding.py
@@ -1105,13 +1105,9 @@ def test_same_same(loaded_model: EmbeddingModule, truncate_input_tokens):
         assert np.allclose(e, combined_vectors[i])
 
     # Next ensuring that the two identical sentences yield identical results (and 3rd does not)
-    assert np.allclose(
-        combined_vectors[0], combined_vectors[1], rtol=1e-05, atol=1e-08
-    )
+    assert np.allclose(combined_vectors[0], combined_vectors[1], rtol=1e-05, atol=1e-08)
     assert not np.array_equal(combined_vectors[1], combined_vectors[2])
-    assert np.allclose(
-        separate_vectors[0], separate_vectors[1], rtol=1e-05, atol=1e-08
-    )
+    assert np.allclose(separate_vectors[0], separate_vectors[1], rtol=1e-05, atol=1e-08)
     assert not np.allclose(
         separate_vectors[1], separate_vectors[2], rtol=1e-05, atol=1e-08
     )