Fix Flax `from_pt` (#1436)

Commit

3 years ago

Fix Flax `from_pt` (#1436) Fix Flax `from_pt`. It worked for models but not for pipelines. Accidentally broken in #1107.