search

Building-ML-Pipelines / building-machine-learning-pipelines

Code repository for the O'Reilly publication "Building Machine Learning Pipelines" by Hannes Hapke & Catherine Nelson
MIT License
583 stars 250 forks source link

The way to extend BaseExampleGenExecutor has changed in version 0.23 #43

Closed asimonw closed 2 years ago

asimonw commented 3 years ago

In your example code for writing a custom component by extending BaseExampleGenExecutor (see Custom_TFX_Components notebook), your ImageToExample function should no longer accept an input_dict explicitly.

This has changed since version 0.23. The input base can now simply be found at exec_properties['input_base']. The code as is will result in an error in more recent versions. Hope this helps keeping this valuable resource (thanks for that, btw!) up to date.

hanneshapke commented 2 years ago

Hi @asimonw,

Thank you for filing this issue! I have updated the example (https://github.com/Building-ML-Pipelines/building-machine-learning-pipelines/commit/3c446b4863d5b2f2a5109bc617eaedec0ec5922d) and upgraded the repo to TFX 1.4.

Thank you again for raising this issue!