Update claude endpoint

run.py

@@ -52,8 +52,7 @@ logger = logging.getLogger("desktopenv.experiment")
 
 # wandb config
 ### set your wandb api key here
-os.environ["WANDB_API_KEY"] = ""
-wandb.login(key=os.environ["WANDB_API_KEY"])
+wandb.login(key=os.get("WANDB_API_KEY", None))
 
 
 def config() -> argparse.Namespace:
@@ -148,7 +147,7 @@ def test(
 
     for domain in tqdm(test_all_meta, desc="Domain"):
         for example_id in tqdm(test_all_meta[domain], desc="Example", leave=False):
-            run = wandb.init(project=f"OSworld-{args.action_space}-{args.observation_type}-{args.model}", group=f"{domain}", 
+            run = wandb.init(project=f"OSworld-{args.action_space}-{args.observation_type}-{args.model}", group=f"{domain}",
                     name=f"{example_id}")
             # example setting
             config_file = os.path.join(args.test_config_base_dir, f"examples/{domain}/{example_id}.json")
@@ -164,7 +163,7 @@ def test(
             # wandb each example config settings
             cfg_args["instruction"] = instruction
             cfg_args["start_time"] = datetime.datetime.now().strftime("%Y:%m:%d-%H:%M:%S")
-            run.config.update(cfg_args)   
+            run.config.update(cfg_args)
 
             example_result_dir = os.path.join(
                 args.result_dir,
@@ -279,10 +278,10 @@ if __name__ == '__main__':
         left_info += f"{domain}: {len(test_file_list[domain])}\n"
     logger.info(f"Left tasks:\n{left_info}")
 
-    # get_result(args.action_space,
-    #     args.model,
-    #     args.observation_type,
-    #     args.result_dir,
-    #     test_all_meta
-    # )
+    get_result(args.action_space,
+        args.model,
+        args.observation_type,
+        args.result_dir,
+        test_all_meta
+    )
     test(args, test_file_list)