add functionality required for this test load_collection & collect

[github-actions]

add functionality required for this test load_collection & collect

https://github.com/apache/incubator-wayang/blob/1e0f9e8166225176fe3022de5fbcce3dbcba96b9/python/src/pywy/tests/train_logistic_test.py#L24

#  limitations under the License.
#

import pytest

from pywy.dataquanta import WayangContext
from pywy.platforms.java import JavaPlugin
from pywy.platforms.spark import SparkPlugin

# TODO: add functionality required for this test load_collection & collect
@pytest.mark.skip(reason="no way of currently testing this, since we are missing implementations for load_collection & collect")
def test_train_and_predict():
    ctx = WayangContext().register({JavaPlugin, SparkPlugin})

    features = ctx.load_collection([[0.0, 1.0], [1.0, 0.0], [1.0, 1.0], [0.0, 0.0]])
    labels = ctx.load_collection([1.0, 1.0, 0.0, 0.0])

    model = features.train_logistic_regression(labels)
    predictions = model.predict(features)

    result = predictions.collect()
    print("Predictions:", result)

    assert len(result) is 4, f"Expected len(result) to be 4, but got: {len(result)}"
    for pred in result:
        assert pred in [0.0, 1.0], f"Expected prediction to be in [0.0, 0.1], but got: {pred}"

f35294bf9a27cc63281f965c1bf55a42419e4b35

[AjayBoddeda4]

Hi, I am Ajay Boddeda, a GSoC 2026 applicant working on the DataFrames API proposal for Apache Wayang.This issue is relevant to my proposal. The load_collection and collect operations are fundamental to any DataFrame API — users need to load data into a DataFrame and collect results back.In the DataFrame API I am designing, load_collection would create a DataFrame from an in-memory collection with an inferred Schema, and collect would return the rows as a list. These are essentially the entry and exit points of the DataFrame pipeline.Implementing these for the Python API would also naturally complement the DataFrame abstraction since Python data scientists expect to work with collections and DataFrames interchangeably.Would love to contribute to this as part of the broader DataFrame API work.