support if dataframe of instance pyspark.sql.connect.dataframe.Dataframe is passed as input

[diviteja-g]

support if dataframe of instance pyspark.sql.connect.dataframe.Dataframe is passed as input

Description

Spark Expectations currently fail on databricks runtime 14 and above when dataframe of instance pyspark.sql.connect.dataframe.Dataframe is passed as input.

Updated logic to check for both dataframe instances as either of it can be passed depending on how dataframe is created (read from table, output of a query, read from files)

Related Issue

#106

Motivation and Context

How Has This Been Tested?

Uploaded wheel file generated to Databricks cluster runtime versions 13, 14, & 15 and ran spark expectations job

Screenshots (if appropriate):

Types of changes

• Bug fix (non-breaking change which fixes an issue)
• New feature (non-breaking change which adds functionality)
• Breaking change (fix or feature that would cause existing functionality to change)

Checklist:

• My code follows the code style of this project.
• My change requires a change to the documentation.
• I have updated the documentation accordingly.
• I have read the CONTRIBUTING document.
• I have added tests to cover my changes.
• All new and existing tests passed.

[stampthecoder]

@diviteja-g this is exactly what we need! The current version lacks this pull requests dataframe types for spark connect. Is this project still active?

[asingamaneni]

LGTM, thanks for your contribution