format

Author: atharvas

src/datasmith/agents/context_synthesis.py

@@ -9,14 +9,14 @@
 from datetime import datetime, timezone
 from pathlib import Path
 
+import docker
 import dspy
+from docker.errors import APIError, ImageNotFound, NotFound
 
-import docker
 from datasmith.agents.tool_executor import ContainerToolExecutor
 from datasmith.docker.context import BuildResult, ContextRegistry, DockerContext
 from datasmith.docker.orchestrator import gen_run_labels
 from datasmith.docker.validation import Task
-from docker.errors import APIError, ImageNotFound, NotFound
 
 logger = logging.getLogger(__name__)
 

src/datasmith/docker/context.py

@@ -16,9 +16,10 @@
 from typing import Any, ClassVar
 
 import docker
+from docker.errors import APIError, DockerException, ImageNotFound
+
 from datasmith.execution.utils import _get_commit_info
 from datasmith.logging_config import get_logger
-from docker.errors import APIError, DockerException, ImageNotFound
 
 logger = get_logger("docker.context")