把 tqdm.autonotebook 换成tqdm.auto

fastNLP/api/utils.py

@@ -20,10 +20,10 @@ except ImportError:
         from urllib.request import urlopen
         from urllib.parse import urlparse
 try:
-    from tqdm import tqdm
-except ImportError:
-    tqdm = None  # defined below
-
+    from tqdm.auto import tqdm
+except:
+    from ..core.utils import _pseudo_tqdm as tqdm
+    
 # matches bfd8deac from resnet18-bfd8deac.pth
 HASH_REGEX = re.compile(r'-([a-f0-9]*)\.')
 

fastNLP/automl/enas_trainer.py

@@ -9,7 +9,7 @@ import numpy as np
 import torch
 
 try:
-    from tqdm.autonotebook import tqdm
+    from tqdm.auto import tqdm
 except:
     from ..core.utils import _pseudo_tqdm as tqdm
 

fastNLP/core/trainer.py

@@ -311,7 +311,7 @@ import torch
 from torch import nn
 
 try:
-    from tqdm.autonotebook import tqdm
+    from tqdm.auto import tqdm
 except:
     from .utils import _pseudo_tqdm as tqdm
 

fastNLP/io/dataset_loader.py

@@ -20,7 +20,10 @@ from .file_reader import _read_csv, _read_json, _read_conll
 
 
 def _download_from_url(url, path):
-    from tqdm import tqdm
+    try:
+        from tqdm.auto import tqdm
+    except:
+        from ..core.utils import _pseudo_tqdm as tqdm
     import requests
     
     """Download file"""

fastNLP/models/enas_trainer.py

@@ -9,7 +9,7 @@ import torch
 import math
 
 try:
-    from tqdm.autonotebook import tqdm
+    from tqdm.auto import tqdm
 except:
     from ..core.utils import _pseudo_tqdm as tqdm