Skip to content

Commit 1f9435b

Browse files
committed
remove some uneeded imports
1 parent f72295c commit 1f9435b

File tree

3 files changed

+3
-8
lines changed

3 files changed

+3
-8
lines changed

eval_protocol/pytest/github_action_rollout_processor.py

Lines changed: 2 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -1,17 +1,13 @@
11
import asyncio
2-
import base64
3-
import json
42
import os
5-
import tempfile
63
import time
7-
import zipfile
84
from typing import Any, Callable, Dict, List, Optional
95

106
import requests
117

12-
from eval_protocol.models import EvaluationRow, Message, Status
8+
from eval_protocol.models import EvaluationRow, Status
139
from eval_protocol.data_loader.dynamic_data_loader import DynamicDataLoader
14-
from eval_protocol.types.remote_rollout_processor import DataLoaderConfig, RolloutMetadata
10+
from eval_protocol.types.remote_rollout_processor import DataLoaderConfig
1511

1612
from .rollout_processor import RolloutProcessor
1713
from .types import RolloutProcessorConfig

tests/pytest/test_utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
import asyncio
2-
from unittest.mock import AsyncMock, MagicMock, patch
2+
from unittest.mock import MagicMock
33
import pytest
44

55
from eval_protocol.pytest.evaluation_test_utils import rollout_processor_with_retry

tests/test_evaluation_postprocess.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,5 @@
11
"""Tests for evaluation postprocess functionality."""
22

3-
import pytest
43
from unittest.mock import Mock, patch
54

65
from eval_protocol.models import EvaluationRow, EvaluateResult, EvalMetadata, ExecutionMetadata, InputMetadata, Message

0 commit comments

Comments
 (0)