Fixed formatting

Signed-off-by: Jim Martens <github@2martens.de>
2019-04-29 11:47:04 +02:00
parent 2d4bd54d07
commit 2735c1c5a9
1 changed files with 2 additions and 1 deletions
--- a/src/twomartens/masterthesis/ssd.py
+++ b/src/twomartens/masterthesis/ssd.py
@ -191,7 +191,8 @@ def _predict_one_epoch(dataset: tf.data.Dataset,

        # save predictions batch-wise to prevent memory problems
        if nr_digits is not None:
-            filename = f"{output_file}-{counter:{nr_digits}d}.npy"
+            counter = str(counter).zfill(nr_digits)
+            filename = f"{output_file}-{counter}.npy"
        else:
            filename = f"{output_file}-{counter:d}.npy"