linting and local testing tidy up

This commit is contained in:
Shannon Sands 2025-05-12 08:07:39 +10:00
parent 141ab66792
commit bfc967c4bd
3 changed files with 242 additions and 310 deletions

View file

@ -3,7 +3,7 @@ import json
import logging
import random
import re
from typing import Any, Dict, List, Optional, Tuple, Union
from typing import Dict, List, Optional, Tuple, Union
from atroposlib.envs.base import BaseEnv, BaseEnvConfig, OpenaiConfig, ScoredDataGroup
from atroposlib.utils.tokenize_for_trainer import tokenize_for_trainer