fix issue with trailing zero in tests

sdv-dev · Oct 27, 2023 · 7c18b14 · 7c18b14
1 parent d1415ba
commit 7c18b14
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/integration/synthesizer/test_tvae.py b/tests/integration/synthesizer/test_tvae.py
@@ -44,7 +44,7 @@ def test_tvae(tmpdir, capsys):
     assert set(loss_values.columns) == {'Epoch', 'Batch', 'Loss'}
     assert all(loss_values['Batch'] == 0)
     last_loss_val = loss_values['Loss'].iloc[-1]
-    assert f'Loss: {round(last_loss_val, 3)}: 100%' in captured_out
+    assert f'Loss: {round(last_loss_val, 3):.3f}: 100%' in captured_out
 
 
 def test_drop_last_false():