Skip to content

Commit d2937ea

Browse files
fix: enable keras and final evaluation make tests (#177)
- run test-keras against the spaceship dataset with keras backend - enable final evaluation in test-pytorch - bump project version to 1.3.1
1 parent 0789b80 commit d2937ea

2 files changed

Lines changed: 28 additions & 6 deletions

File tree

Makefile

Lines changed: 27 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -189,15 +189,37 @@ test-pytorch: build
189189
--max-iterations 2 \
190190
--work-dir /workdir/test_pytorch_house_prices \
191191
--spark-mode local \
192-
--allowed-model-types pytorch
192+
--allowed-model-types pytorch \
193+
--enable-final-evaluation
193194
@echo "✅ PyTorch test passed!"
194195

195-
# Test Keras specifically (requires appropriate dataset)
196+
# Test Keras specifically
196197
.PHONY: test-keras
197198
test-keras: build
198-
@echo "⚠️ Keras requires IMAGE_PATH or TEXT_STRING data layout"
199-
@echo "⚠️ Spaceship Titanic is FLAT_NUMERIC - incompatible with Keras"
200-
@echo "Skipping Keras test (add image/text dataset to enable)"
199+
@echo "🧪 Testing Keras model type..."
200+
docker run --rm \
201+
--add-host=host.docker.internal:host-gateway \
202+
$(CONFIG_MOUNT) \
203+
$(CONFIG_ENV) \
204+
-v $(PWD)/examples/datasets:/data:ro \
205+
-v $(PWD)/workdir:/workdir \
206+
-e OPENAI_API_KEY=$(OPENAI_API_KEY) \
207+
-e ANTHROPIC_API_KEY=$(ANTHROPIC_API_KEY) \
208+
-e SPARK_LOCAL_CORES=4 \
209+
-e SPARK_DRIVER_MEMORY=4g \
210+
-e KERAS_BACKEND=tensorflow \
211+
plexe:py$(PYTHON_VERSION) \
212+
python -m plexe.main \
213+
--train-dataset-uri /data/spaceship-titanic/train.parquet \
214+
--user-id test_user \
215+
--intent "predict whether a passenger was transported" \
216+
--experiment-id test_keras \
217+
--max-iterations 2 \
218+
--work-dir /workdir/test_keras \
219+
--spark-mode local \
220+
--allowed-model-types keras \
221+
--enable-final-evaluation
222+
@echo "✅ Keras test passed!"
201223

202224
# Test all model types (let agents explore)
203225
.PHONY: test-all-models

pyproject.toml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
[tool.poetry]
22
name = "plexe"
3-
version = "1.3.0"
3+
version = "1.3.1"
44
description = "An agentic framework for building ML models from natural language"
55
authors = [
66
"Marcello De Bernardi <mdebernardi@plexe.ai>",

0 commit comments

Comments
 (0)