Skip to content

Commit 40cab67

Browse files
removed tqdm
1 parent f691703 commit 40cab67

File tree

1 file changed

+2
-5
lines changed

1 file changed

+2
-5
lines changed

Diff for: pyLOM/NN/utils.py

+2-5
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,6 @@
1313
from torch import Generator, randperm, default_generator
1414
from itertools import product, accumulate
1515
from typing import List, Optional, Tuple, cast, Sequence, Union, Callable
16-
from tqdm import tqdm
1716

1817
from . import DEVICE
1918
from ..utils.cr import cr
@@ -688,8 +687,7 @@ def map(self, function: Callable, fn_kwargs: dict = {}, batched: bool = False, b
688687

689688
if not batched:
690689
batch_size = 1
691-
pbar = tqdm(range(0, len(self), batch_size), desc="Mapping dataset", unit="iters")
692-
for i in pbar:
690+
for i in range(0, len(self), batch_size):
693691
batch = self[i:i + batch_size]
694692
if self.variables_in is not None:
695693
inputs, outputs = batch
@@ -733,8 +731,7 @@ def filter(
733731
if not batched:
734732
batch_size = 1
735733
indices = []
736-
pbar = tqdm(range(0, len(self), batch_size), desc="Filtering dataset", unit="iters")
737-
for i in pbar:
734+
for i in range(0, len(self), batch_size):
738735
batch = self[i:i + batch_size]
739736
if self.variables_in is not None:
740737
inputs, outputs = batch

0 commit comments

Comments
 (0)