diff --git a/tests/unit/torch/examples/test_01_getting_started.py b/tests/unit/torch/examples/test_01_getting_started.py index d0ae8dc5f9..2dcaea452a 100644 --- a/tests/unit/torch/examples/test_01_getting_started.py +++ b/tests/unit/torch/examples/test_01_getting_started.py @@ -46,6 +46,6 @@ def test_example_01_getting_started(tb): "val_binary_accuracy", "val_binary_auroc", "val_binary_precision", - "val_binary_recall" + "val_binary_recall", ] )