diff --git a/environments/tool_use_multiturn_server.py b/environments/tool_use_multiturn_server.py index 96744523..ce840de4 100644 --- a/environments/tool_use_multiturn_server.py +++ b/environments/tool_use_multiturn_server.py @@ -26,11 +26,11 @@ import re from collections import Counter from typing import Dict, List, Optional, Tuple +import wandb from datasets import load_dataset from pydantic import Field from tqdm.asyncio import tqdm_asyncio -import wandb from atroposlib.envs.base import ( APIServerConfig, BaseEnv,