use tqdm.auto

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

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

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