lp4.py

# Store the variable as an artifact
train_artifact = lineapy.save(train_features, "train_data")

# Check object type
print(type(train_artifact))

# Store the variable as an artifact
train_labels = lineapy.save(train_labels, "train_labels")

# Check object type
print(type(train_labels))

# Store the variable as an artifact
test_artifact = lineapy.save(test_features, "test_data")

# Check object type
print(type(test_artifact))

# Store the variable as an artifact
test_labels = lineapy.save(test_labels, "test_labels")

# Check object type
print(type(test_labels))