fix: add support for PyDataFrame extraction in PySchema implementation

Author: kosiew

src/catalog.rs

@@ -15,6 +15,7 @@
 // specific language governing permissions and limitations
 // under the License.
 
+use crate::dataframe::PyDataFrame;
 use crate::dataset::Dataset;
 use crate::errors::{py_datafusion_err, to_datafusion_err, PyDataFusionError, PyDataFusionResult};
 use crate::table::PyTableProvider;
@@ -203,6 +204,12 @@ impl PySchema {
             py_table.table
         } else if let Ok(py_provider) = table_provider.extract::<PyTableProvider>() {
             py_provider.into_inner()
+        } else if table_provider.extract::<PyDataFrame>().is_ok() {
+            return Err(PyDataFusionError::Common(
+                "Expected a Table or TableProvider. Convert DataFrames with \"DataFrame.into_view()\" or \"TableProvider.from_dataframe()\"."
+                    .to_string(),
+            )
+            .into());
         } else if let Some(provider) = table_provider_from_pycapsule(&table_provider)? {
             provider
         } else {