CreativeEngineer commited on
Commit
798165c
·
1 Parent(s): 904b958

Allow class definitions in safe_builtins

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -328,6 +328,7 @@ def verify_perf_takehome_code(code: str, seed: int = 123) -> dict:
328
  "abs": abs,
329
  "all": all,
330
  "any": any,
 
331
  "dict": dict,
332
  "enumerate": enumerate,
333
  "int": int,
@@ -335,6 +336,7 @@ def verify_perf_takehome_code(code: str, seed: int = 123) -> dict:
335
  "list": list,
336
  "max": max,
337
  "min": min,
 
338
  "range": range,
339
  "sum": sum,
340
  "tuple": tuple,
@@ -342,6 +344,7 @@ def verify_perf_takehome_code(code: str, seed: int = 123) -> dict:
342
  }
343
  exec_globals = {
344
  "__builtins__": safe_builtins,
 
345
  "KernelBuilder": KernelBuilder,
346
  "HASH_STAGES": HASH_STAGES,
347
  "VLEN": VLEN,
 
328
  "abs": abs,
329
  "all": all,
330
  "any": any,
331
+ "__build_class__": __build_class__,
332
  "dict": dict,
333
  "enumerate": enumerate,
334
  "int": int,
 
336
  "list": list,
337
  "max": max,
338
  "min": min,
339
+ "object": object,
340
  "range": range,
341
  "sum": sum,
342
  "tuple": tuple,
 
344
  }
345
  exec_globals = {
346
  "__builtins__": safe_builtins,
347
+ "__name__": "__main__",
348
  "KernelBuilder": KernelBuilder,
349
  "HASH_STAGES": HASH_STAGES,
350
  "VLEN": VLEN,