Merge pull request #414 from pcattori/fix-tuc-pandas

Fix errors related to TUC pandas usage

tamr_unify_client/dataset/resource.py

@@ -1,6 +1,7 @@
 from copy import deepcopy
+import os
+from typing import TYPE_CHECKING
 
-import pandas as pd
 import simplejson as json
 
 from tamr_unify_client.attribute.collection import AttributeCollection
@@ -11,6 +12,10 @@
 from tamr_unify_client.dataset.usage import DatasetUsage
 from tamr_unify_client.operation import Operation
 
+BUILDING_DOCS = os.environ.get("TAMR_CLIENT_DOCS") == "1"
+if TYPE_CHECKING or BUILDING_DOCS:
+    import pandas as pd
+
 
 class Dataset(BaseResource):
     """A Tamr dataset."""
@@ -86,7 +91,7 @@ def _update_records(self, updates, **json_args):
         )
 
     def upsert_from_dataframe(
-        self, df: pd.DataFrame, *, primary_key_name: str, ignore_nan: bool = True
+        self, df: "pd.DataFrame", *, primary_key_name: str, ignore_nan: bool = True
     ) -> dict:
         """Upserts a record for each row of `df` with attributes for each column in `df`.