@@ -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
197198test-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
0 commit comments