nits

Signed-off-by: David Korczynski <[email protected]>

run_all_experiments.py

@@ -26,7 +26,7 @@
 from multiprocessing import Pool, Process
 from typing import Any
 
-import google.cloud.logging
+from google.cloud import logging as cloud_logging
 
 import run_one_experiment
 from data_prep import introspector
@@ -36,9 +36,10 @@
 from llm_toolkit import models, prompt_builder
 
 try:
-  client = google.cloud.logging.Client()
+  client = cloud_logging.Client()
   client.setup_logging()
-except google.auth.exceptions.DefaultCredentialsError:
+except Exception as e:
+  # For local runs we continue
   pass
 
 logger = logging.getLogger(__name__)
@@ -523,9 +524,6 @@ def main():
 
   args = parse_args()
   _setup_logging(args.log_level)
-  logging.info('Starting experiments on PR branch')
-  logging.warning('Warning from logging')
-  logging.error('Error from exp')
   logger.info('Starting experiments on PR branch')
 
   # Capture time at start