From bab7adbc7e31d53e730c84b7fa8f2cf5d8b7861e Mon Sep 17 00:00:00 2001 From: "Dr. Kashif Rasul" Date: Fri, 31 Jan 2020 17:22:49 +0100 Subject: [PATCH] Revert "use tqdm.auto" This reverts commit ffb839dcd791152922d5c07fa1b0338d9ffbe67f. --- pts/dataset/stat.py | 2 +- pts/evaluation/evaluator.py | 3 ++- pts/trainer.py | 2 +- 3 files changed, 4 insertions(+), 3 deletions(-) diff --git a/pts/dataset/stat.py b/pts/dataset/stat.py index cb1f172..9f81566 100644 --- a/pts/dataset/stat.py +++ b/pts/dataset/stat.py @@ -3,7 +3,7 @@ from collections import defaultdict from typing import Any, List, NamedTuple, Optional, Set import numpy as np -from tqdm.auto import tqdm +from tqdm import tqdm from .common import FieldName from pts.exception import assert_pts diff --git a/pts/evaluation/evaluator.py b/pts/evaluation/evaluator.py index 047a996..5c696aa 100644 --- a/pts/evaluation/evaluator.py +++ b/pts/evaluation/evaluator.py @@ -10,11 +10,12 @@ from typing import ( Union, Callable, ) -from tqdm.auto import tqdm +from tqdm import tqdm # Third-party imports import numpy as np import pandas as pd +from tqdm import tqdm from pts.model import Quantile, Forecast from pts.feature import get_seasonality diff --git a/pts/trainer.py b/pts/trainer.py index 2d7052b..00728d8 100644 --- a/pts/trainer.py +++ b/pts/trainer.py @@ -4,7 +4,7 @@ from typing import Any, List, NamedTuple, Optional, Union import torch import torch.nn as nn from torch.utils.tensorboard import SummaryWriter -from tqdm.auto import tqdm +from tqdm import tqdm from torch.utils.data import DataLoader