tease 0.2.0

Rapidly prototype interfaces for your machine learning models
Documentation 
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
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84

#!/usr/bin/env python3
"""Verify the crate archive, then compile and run an isolated headless consumer."""
import json
import os
from pathlib import Path
import subprocess
import tarfile
import tempfile

root = Path(__file__).resolve().parents[1]
metadata = json.loads(subprocess.check_output(
    ["cargo", "metadata", "--no-deps", "--format-version", "1"], cwd=root))
package = next(p for p in metadata["packages"] if Path(p["manifest_path"]).parent == root)
name = f'{package["name"]}-{package["version"]}'
target = Path(metadata["target_directory"])
subprocess.run(["cargo", "package", "--locked", "--allow-dirty"], cwd=root, check=True)
archive_path = target / "package" / f"{name}.crate"

with tempfile.TemporaryDirectory(prefix="tease-package-") as scratch:
    scratch = Path(scratch)
    unpacked = scratch / "unpacked"
    with tarfile.open(archive_path) as archive:
        members = archive.getmembers()
        names = {member.name for member in members}
        required = ["Cargo.toml", "Cargo.lock", "LICENSE-MIT", "LICENSE-APACHE", "README.md",
                    "CHANGELOG.md", "RELEASING.md", "docs/migration-0.2.md", "docs/examples.md",
                    "src/ui.js", "src/worker.rs", "examples/smartcore_classification.rs"]
        missing = [path for path in required if f"{name}/{path}" not in names]
        if missing:
            raise RuntimeError(f"Package missing required files: {missing}")
        for member in members:
            destination = (unpacked / member.name).resolve()
            if not destination.is_relative_to(unpacked.resolve()):
                raise RuntimeError(f"Invalid archive path: {member.name}")
            if member.isdir():
                destination.mkdir(parents=True, exist_ok=True)
            elif member.isfile():
                destination.parent.mkdir(parents=True, exist_ok=True)
                with archive.extractfile(member) as source:
                    destination.write_bytes(source.read())
            else:
                raise RuntimeError(f"Unsupported archive member: {member.name}")
        if any(f"/{part}/" in path for path in names for part in ["assets", "doc_chunks", "bootstrap", ".github", "target"]):
            raise RuntimeError("Package contains obsolete assets or build/CI artifacts")

    consumer = scratch / "consumer"
    (consumer / "src").mkdir(parents=True)
    (consumer / "Cargo.toml").write_text(
        '[package]\nname = "tease-package-smoke"\nversion = "0.0.0"\n'
        'edition = "2021"\nrust-version = "1.88"\n\n[dependencies]\n'
        f'tease = {{ path = {json.dumps(str(unpacked / name))} }}\n')
    (consumer / "src" / "main.rs").write_text('''
use std::{cell::Cell, rc::Rc};
use tease::{Choice, Input, Output, Teaser, Value};
fn main() -> Result<(), tease::Error> {
    let simple = Teaser::default().with_function(|x| x.iter().sum()).prepare()?;
    assert_eq!(simple.predict(&[Value::Number(3.0)])?, vec![Value::Number(3.0)]);
    let session = Teaser::default()
        .with_inputs(vec![Input::number("Value", 2.0), Input::Dropdown {
            label: Some("Category".into()), initial_value: 0,
            options: vec![Choice::text("Alpha", "alpha")],
        }])
        .with_outputs(vec![Output::number("Value", 2), Output::text("Category")])
        .with_predictor_factory(|| {
            let calls = Rc::new(Cell::new(0));
            Ok(move |values: &[Value]| {
                calls.set(calls.get() + 1);
                let number = values[0].as_number()?;
                if number < 0.0 { return Err("Negative value.".into()); }
                Ok(vec![Value::Number(number), Value::Text(values[1].as_text()?.into())])
            })
        }).prepare()?;
    assert!(session.predict(&[Value::Number(-1.0), Value::Text("alpha".into())]).is_err());
    assert_eq!(session.predict(&[Value::Number(2.0), Value::Text("alpha".into())])?,
        vec![Value::Number(2.0), Value::Text("alpha".into())]);
    println!("Packaged Tease consumer passed.");
    Ok(())
}
''')
    env = os.environ.copy()
    env["CARGO_TARGET_DIR"] = str(target / "package-consumer")
    subprocess.run(["cargo", "run", "--manifest-path", str(consumer / "Cargo.toml")],
                   cwd=consumer, env=env, check=True)
print(f"Verified {archive_path.name}: {len(names)} files and an isolated consumer.")