From 2432ac16c34fb00ae9789cb0ce84939dda5fe46a Mon Sep 17 00:00:00 2001 From: Patrick Devaney Date: Mon, 23 Dec 2024 01:44:26 -0500 Subject: [PATCH] more linting tests --- new_features_examples/multi_tool_usage_agent.py | 2 +- tests/agent_evals/auto_test_eval.py | 12 ++++++------ 2 files changed, 7 insertions(+), 7 deletions(-) diff --git a/new_features_examples/multi_tool_usage_agent.py b/new_features_examples/multi_tool_usage_agent.py index 1af421e2..33c61abe 100644 --- a/new_features_examples/multi_tool_usage_agent.py +++ b/new_features_examples/multi_tool_usage_agent.py @@ -19,7 +19,7 @@ class ToolDefinition: callable: Optional[Callable] = None -def extract_type_hints(func: Callable) -> Dict[str, Any]: +def extract_type_hints(func: Callable) -> dict: """Extract parameter types from function type hints.""" return typing.get_type_hints(func) diff --git a/tests/agent_evals/auto_test_eval.py b/tests/agent_evals/auto_test_eval.py index 43205ecb..e2bc06f7 100644 --- a/tests/agent_evals/auto_test_eval.py +++ b/tests/agent_evals/auto_test_eval.py @@ -93,13 +93,13 @@ class SwarmsIssueReporter: self.last_issue_time = datetime.now() def _get_swarms_version(self) -> str: - """Get the installed version of Swarms.""" - try: - import swarms + """Get the installed version of Swarms.""" + try: + import swarms - return swarms.__version__ - except: - return "Unknown" + return swarms.__version__ + except: + return "Unknown" def _get_gpu_info(self) -> Tuple[bool, Optional[str]]: """Get GPU information and CUDA availability."""