style: fix pre-commit issues

- Fix trailing whitespace
- Fix import ordering with isort
- Apply autopep8 formatting

Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
pull/1157/head
Michael Walker 7 months ago
parent bb116151e7
commit 7b4956fc90

@ -8,51 +8,53 @@ Validates that all 11 new Agentcore icons can be imported and used correctly.
from diagrams import Diagram
from diagrams.aws.ml import (
Agentcore,
AiAgent,
Runtime,
AiAgent,
BrowserTool,
CodeInterpreter,
Evaluations,
Gateway,
Identity,
CodeInterpreter,
Observability,
BrowserTool,
Memory,
Evaluations,
PolicyEngineAgenticGuardrails
Observability,
PolicyEngineAgenticGuardrails,
Runtime,
)
def test_agentcore_icons():
"""Test all Agentcore icons in a sample diagram"""
with Diagram("AWS Agentcore Bedrock Icons Test", show=False, filename="agentcore_test"):
# Core Agentcore components
agentcore = Agentcore("Agentcore")
ai_agent = AiAgent("AI Agent")
runtime = Runtime("Runtime")
gateway = Gateway("Gateway")
# Identity and security
identity = Identity("Identity")
policy_engine = PolicyEngineAgenticGuardrails("Policy Engine")
# Tools and capabilities
code_interpreter = CodeInterpreter("Code Interpreter")
browser_tool = BrowserTool("Browser Tool")
memory = Memory("Memory")
# Monitoring and evaluation
observability = Observability("Observability")
evaluations = Evaluations("Evaluations")
# Create a flow showing the relationships
agentcore >> ai_agent >> runtime
ai_agent >> [code_interpreter, browser_tool, memory]
gateway >> identity >> policy_engine
[runtime, code_interpreter, browser_tool] >> observability
observability >> evaluations
print("✅ Test diagram created successfully!")
print("📁 Generated file: agentcore_test.png")
print("🔍 Please verify the icons render correctly in the diagram")
if __name__ == "__main__":
test_agentcore_icons()

@ -8,15 +8,16 @@ Tests Amazon Nova, AWS App Studio, Amazon CodeWhisperer, and AWS Neuron
from diagrams import Diagram
from diagrams.aws.ml import AmazonNova, AWSAppStudio, AWSNeuron
def test_nova_icons():
"""Test the new Nova and AI service icons"""
with Diagram("New AWS AI/ML Services", show=False, filename="nova_test"):
# Create instances of the new services
nova = AmazonNova("Amazon Nova")
app_studio = AWSAppStudio("AWS App Studio")
neuron = AWSNeuron("AWS Neuron")
# Create a simple flow
nova >> app_studio >> neuron
@ -24,5 +25,6 @@ def test_nova_icons():
print("📁 Generated file: nova_test.png")
print("🔍 Please verify the icons render correctly in the diagram")
if __name__ == "__main__":
test_nova_icons()

Loading…
Cancel
Save