Rename cst visitor class for clarity

huggingface · Oct 25, 2024 · e7fd29e · e7fd29e
1 parent 1158270
commit e7fd29e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/utils/check_task_parameters.py b/utils/check_task_parameters.py
@@ -159,7 +159,7 @@ def visit_ImportFrom(self, node: cst.ImportFrom):
                 self.type_to_module[alias.name.value] = module_name
 
 
-class MethodParameterCollector(cst.CSTVisitor):
+class MethodArgumentsCollector(cst.CSTVisitor):
     """Collects parameter types and docstrings from a method."""
 
     def __init__(self, method_name: str):
@@ -528,7 +528,7 @@ def _check_parameters(
     parameters_module.visit(params_collector)
     dataclass_params = params_collector.parameters
     # Get existing parameters from the method
-    method_collector = MethodParameterCollector(method_name)
+    method_collector = MethodArgumentsCollector(method_name)
     inference_client_module.visit(method_collector)
     existing_params = method_collector.parameters