From c3f14d0d397094101b3516f1aa654b7693760822 Mon Sep 17 00:00:00 2001
From: dcarron <daniel.carron@idiap.ch>
Date: Wed, 14 Jun 2023 14:06:08 +0200
Subject: [PATCH] Fixed progress bar when loading samples
---
src/ptbench/data/dataset.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/src/ptbench/data/dataset.py b/src/ptbench/data/dataset.py
index 07f18acf..5966f392 100644
--- a/src/ptbench/data/dataset.py
+++ b/src/ptbench/data/dataset.py
@@ -10,9 +10,9 @@ import pathlib
import random
import torch
-import tqdm
from torchvision.transforms import RandomRotation
+from tqdm import tqdm
RANDOM_ROTATION = [RandomRotation(15)]
"""Shared data augmentation based on random rotation only."""
@@ -179,7 +179,7 @@ class JSONDataset:
dict(zip(self.fieldnames, k)),
self.post_transforms,
)
- for n, k in tqdm.tqdm(enumerate(samples))
+ for n, k in enumerate(tqdm(samples))
]
return retval
--
GitLab