Update evaluator.py

Author: codelion

examples/mlx_metal_kernel_opt/evaluator.py

@@ -152,11 +152,17 @@ def _extract_custom_attention_class(self, program_text: str) -> Optional[Any]:
                 actual_program_text = program_text
 
             # Create execution environment
+            import math
+            import numpy as np
+            import time
+            from typing import Optional, Tuple, Any
+            
             exec_globals = {
                 "__builtins__": __builtins__,
                 "mx": mx,
                 "nn": nn,
                 "np": np,
+                "math": math,
                 "time": time,
                 "Optional": Optional,
                 "Tuple": Tuple,