Gergő Jedlicska
46 lines
1.2 KiB
Python
46 lines
1.2 KiB
Python
"""
|
|
This module extracts the JSON schema from the FunctionInputs class the notebook.
|
|
|
|
WARNING: you probably should not be modifying this unless you know what you are doing.
|
|
"""
|
|
import json
|
|
from pathlib import Path
|
|
|
|
import papermill as pm
|
|
|
|
|
|
def save_schema_to_file(file_path: str):
|
|
"""
|
|
Executes the notebook without valid input values.
|
|
|
|
Its fine, since we're only interested in the result of the `function_inputs` cell.
|
|
"""
|
|
notebook_path = "function.ipynb"
|
|
run_result = pm.execute_notebook(notebook_path, "-")
|
|
|
|
# find the cell by its tag
|
|
function_inputs_cells = [
|
|
c for c in run_result.cells if "function_inputs" in c.metadata.tags
|
|
]
|
|
|
|
# should have only found 1
|
|
if len(function_inputs_cells) != 1:
|
|
msg = (
|
|
"Expected 1 cell to have the tag `function_inputs`,"
|
|
f" found {len(function_inputs_cells)}"
|
|
)
|
|
raise ValueError(msg)
|
|
|
|
# parse and dump the schema to make sure its a valid json object
|
|
schema = json.dumps(json.loads(function_inputs_cells[0].outputs[0].text))
|
|
|
|
Path(file_path).write_text(schema)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
import sys
|
|
|
|
_, file_path = sys.argv
|
|
|
|
save_schema_to_file(file_path)
|