Function runPythonTensorSnippet

tests/test_python_integration.cpp:312–356

    }

    PythonTensorResult runPythonTensorSnippet(const std::string& script) {
        const lfs::python::GilAcquire gil;

        PyObject* globals = PyDict_New();
        PyObject* locals = PyDict_New();
        if (!globals || !locals) {
            Py_XDECREF(globals);
            Py_XDECREF(locals);
            throw std::runtime_error("Failed to allocate Python dictionaries");
        }

        PyDict_SetItemString(globals, "__builtins__", PyEval_GetBuiltins());

        PyObject* exec_result = PyRun_String(script.c_str(), Py_file_input, globals, locals);
        if (!exec_result) {
            const auto error = consumePythonError();
            Py_DECREF(globals);
            Py_DECREF(locals);
            throw std::runtime_error(error);
        }
        Py_DECREF(exec_result);

        auto* const shape_obj = PyDict_GetItemString(locals, "result_shape");
        auto* const values_obj = PyDict_GetItemString(locals, "result_values");
        if (!shape_obj || !PyTuple_Check(shape_obj) || !values_obj || !PyList_Check(values_obj)) {
            Py_DECREF(globals);
            Py_DECREF(locals);
            throw std::runtime_error("Python snippet did not populate result_shape/result_values");
        }

        PythonTensorResult result;
        result.shape.reserve(static_cast<size_t>(PyTuple_Size(shape_obj)));
        for (Py_ssize_t i = 0; i < PyTuple_Size(shape_obj); ++i) {
            result.shape.push_back(PyLong_AsLongLong(PyTuple_GetItem(shape_obj, i)));
        }

        result.values.reserve(static_cast<size_t>(PyList_Size(values_obj)));
        for (Py_ssize_t i = 0; i < PyList_Size(values_obj); ++i) {
            result.values.push_back(static_cast<float>(PyFloat_AsDouble(PyList_GetItem(values_obj, i))));
        }

        Py_DECREF(globals);
        Py_DECREF(locals);
        return result;
    }

    std::vector<long long> runPythonHookContextSnippet(const std::string& registration_script,
                                                       const lfs::training::HookContext& snapshot_ctx,