tqdm import to pandas pattern

googleapis · Mar 27, 2019 · 351039a · 351039a
1 parent b23d581
commit 351039a
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 1,887 deletions.
diff --git a/bigquery/google/cloud/bigquery/table.py b/bigquery/google/cloud/bigquery/table.py
@@ -30,6 +30,11 @@
 except ImportError:  # pragma: NO COVER
     pandas = None
 
+try:
+    import tqdm
+except ImportError:  # pragma: NO COVER
+    tqdm = None
+
 from google.api_core.page_iterator import HTTPIterator
 
 import google.cloud._helpers
@@ -1336,11 +1341,15 @@ def _to_dataframe_tabledata_list(self, dtypes):
         frames = []
 
         # report progress if tqdm installed
-        try:
-            from tqdm import tqdm
-            pbar = tqdm(desc="Downloading", total=self.total_rows, unit="rows")
-        except (ImportError, KeyError, TypeError):
-            pbar = None
+        pbar = None
+        if tqdm is not None:
+            try:
+                pbar = tqdm.tqdm(
+                    desc="Downloading", total=self.total_rows, unit="rows"
+                )
+            except (KeyError, TypeError):
+                # tqdm error
+                pass
 
         for page in iter(self.pages):
             frames.append(self._to_dataframe_dtypes(page, column_names, dtypes))