* remove infer_schema and update tests

* fixed missing test

* updates based on PR feedback

* make schema non-optional in lineage.py

* fix merge issue

* parameter fix