mirror of
https://github.com/zama-ai/concrete.git
synced 2026-02-09 03:55:04 -05:00
24 lines
610 B
Python
24 lines
610 B
Python
import json
|
|
import sys
|
|
|
|
from pathlib import Path
|
|
|
|
|
|
def main():
|
|
path_to_glob = Path(sys.argv[1])
|
|
notebooks = path_to_glob.glob("*.ipynb")
|
|
|
|
for notebook_file in notebooks:
|
|
with open(notebook_file, "r") as f:
|
|
notebook_dict = json.load(f)
|
|
execution = notebook_dict["metadata"].get("execution", {})
|
|
execution["timeout"] = 1000
|
|
notebook_dict["metadata"]["execution"] = execution
|
|
|
|
with open(notebook_file, "w", newline="\n") as f:
|
|
json.dump(notebook_dict, f, indent=1, ensure_ascii=False)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|