File size: 1,729 Bytes
35e2bb9 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 | """Runs example requests through Odyssey_PyPro_10M, prints the generated code, and executes it on a sample list.
python examples.py
"""
import os
import sys
sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))
from odyssey_pypro import REJECT, generate, load # noqa: E402
EXAMPLES = [
('Write a function named "top" with parameter "xs". Keep numbers at least 3, then sort them from largest to smallest.',
[5, 1, 3, 9, 2, 3]),
('Create a function called "clean" that takes a list "vals". Remove duplicate numbers, then return how many numbers there are.',
[4, 4, 2, 7, 2]),
('Given a list called "data", write a function named "shift" that does this: Add -2 to each number.', [1, 2, 3]),
('Square each number, then return the sum of the numbers. Name the function "sq_sum" and its parameter "nums".', [1, -2, 3]),
('Using the input list "seq", implement the function "head3". Take the first 3 values.', [9, 8, 7, 6, 5]),
('Define "low", a function whose argument is "arr". Keep only values below 0, then find the minimum value.', [3, -4, 0, -1]),
('Write a function named "med" with parameter "xs". Compute the median of the numbers.', [1, 2, 3]), # unsupported
]
if __name__ == "__main__":
model = load()
for text, sample in EXAMPLES:
src = generate(model, text)
print("REQUEST:", text)
print(src, end="")
if src != REJECT:
ns = {}
exec(src, ns) # the network's output is ordinary Python; run it only if you trust this example list
fn = [v for k, v in ns.items() if callable(v) and not k.startswith("__")][0]
print(f"# {fn.__name__}({sample}) -> {fn(list(sample))}")
print()
|