diff --git a/src/bedrock_agentcore/evaluation/runner/dataset_providers.py b/src/bedrock_agentcore/evaluation/runner/dataset_providers.py index 081d626e..89240f14 100644 --- a/src/bedrock_agentcore/evaluation/runner/dataset_providers.py +++ b/src/bedrock_agentcore/evaluation/runner/dataset_providers.py @@ -4,7 +4,7 @@ from abc import ABC, abstractmethod from typing import Any, Dict, List, Optional -import requests + from bedrock_agentcore.evaluation.dataset_client import DatasetClient @@ -130,6 +130,7 @@ def get_dataset(self) -> Dataset: raise ValueError(f"Dataset {self._dataset_id} has no downloadUrl. Status: {response.get('status')}") try: + import requests r = requests.get(download_url, timeout=60, stream=True) r.raise_for_status() except requests.RequestException as e: