Skip to content

GitLab

Commit c2e76e79 authored by baberabb's avatar baberabb
Browse files

Merge remote-tracking branch 'origin/big-refactor' into big-refactor_hgtest

parents f0aa2c7b 1c5a73c9
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 2 deletions
lm_eval/tasks/realtoxicityprompts/metric.py
View file @ c2e76e79
...@@ -9,8 +9,10 @@ from lm_eval.logger import eval_logger ...@@ -9,8 +9,10 @@ from lm_eval.logger import eval_logger
try: try:
API_KEY = os.environ["PERSPECTIVE_API_KEY"] API_KEY = os.environ["PERSPECTIVE_API_KEY"]
except KeyError: except KeyError:
eval_logger.error("PERSPECTIVE_API_KEY is not set") API_KEY = None
raise SystemExit(0) eval_logger.warning(
"PERSPECTIVE_API_KEY is not set. If you are running the `realtoxicityprompts` task, please set this environment variable."
)
url = f"https://commentanalyzer.googleapis.com/v1alpha1/comments:analyze?key={API_KEY}" url = f"https://commentanalyzer.googleapis.com/v1alpha1/comments:analyze?key={API_KEY}"
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment