From ffb839dcd791152922d5c07fa1b0338d9ffbe67f Mon Sep 17 00:00:00 2001
From: "Dr. Kashif Rasul" <kashif.rasul@gmail.com>
Date: Fri, 31 Jan 2020 17:09:06 +0100
Subject: [PATCH] use tqdm.auto

---
 pts/dataset/stat.py         | 2 +-
 pts/evaluation/evaluator.py | 3 +--
 pts/trainer.py              | 2 +-
 3 files changed, 3 insertions(+), 4 deletions(-)

diff --git a/pts/dataset/stat.py b/pts/dataset/stat.py
index 9f81566..cb1f172 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 import tqdm
+from tqdm.auto 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 5c696aa..047a996 100644
--- a/pts/evaluation/evaluator.py
+++ b/pts/evaluation/evaluator.py
@@ -10,12 +10,11 @@ from typing import (
     Union,
     Callable,
 )
-from tqdm import tqdm
+from tqdm.auto 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 00728d8..2d7052b 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 import tqdm
+from tqdm.auto import tqdm
 
 from torch.utils.data import DataLoader