Update quantize.py

mikekgfb · web-flow · commit bb3c3cdf7d3c · 2024-10-11T18:41:25.000-07:00
Fix arg order (args with default after args w/o default)
diff --git a/torchchat/utils/quantize.py b/torchchat/utils/quantize.py
@@ -73,7 +73,7 @@ def get_named_parameters(func: Callable) -> List[str]:
     ]
     return named_params
 
-def validate_args(named_params: List[str], quantizer: Optional[str] = None, q_kwargs: Dict[str, Any]) -> Dict:
+def validate_args(named_params: List[str], q_kwargs: Dict[str, Any], quantizer: Optional[str] = None) -> Dict[str, Any]:
     named_params = 
     for key in q_kwargs.keys():
         if key not in q_kwargs:

Original file line number	Diff line number	Diff line change
`@@ -73,7 +73,7 @@ def get_named_parameters(func: Callable) -> List[str]:`
`73`	`73`	`]`
`74`	`74`	`return named_params`
`75`	`75`
`76`		`-def validate_args(named_params: List[str], quantizer: Optional[str] = None, q_kwargs: Dict[str, Any]) -> Dict:`
	`76`	`+def validate_args(named_params: List[str], q_kwargs: Dict[str, Any], quantizer: Optional[str] = None) -> Dict[str, Any]:`
`77`	`77`	`named_params =`
`78`	`78`	`for key in q_kwargs.keys():`
`79`	`79`	`if key not in q_kwargs:`