more linting tests

kyegomez · Dec 23, 2024 · 2432ac1 · 2432ac1
1 parent 50476b9
commit 2432ac1
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 7 deletions.
diff --git 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
@@ -93,13 +93,13 @@ def __init__(
         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."""