better error messaging around perspective api clien (#3784)

facebookresearch · Jul 13, 2021 · f0a37fe · f0a37fe
1 parent 7ff5a4a
commit f0a37fe
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/projects/safety_bench/utils/perspective_api.py b/projects/safety_bench/utils/perspective_api.py
@@ -72,6 +72,13 @@ def get_perspective_api_key(opt) -> Optional[str]:
 
     if api_key is None:
         logging.warning("No Perspective API key is available.")
+    else:
+        assert PERSPECTIVE_CLIENT_AVAILABLE, (
+            "You must install the perspective API client. "
+            "Please run:\n`pip install google-api-python-client`"
+            "\nor follow the instructions here:\n"
+            "<https://github.com/googleapis/google-api-python-client>"
+        )
 
     opt["perspective_api_key"] = api_key