awslabs / awslabs/python-deequ

Lambda arguments are not recognized (Sagemaker)

Open
#54 3 comments 0 reactions 0 assignees View on GitHub
bug
Dominant language
Jupyter Notebook
Stars
826
Forks
158
Avg merge
9d 22h
Merged PRs (30d)
3

Description

**Describe the bug**
When passing a lambda function for an assertion for hasSize, hasMin, or hasMax, it results in a "Can't execute the assertion" error.

**To Reproduce**
Steps to reproduce the behavior:
Use any constraint that requires a lambda function

I used the one listed in the GitHub tutorials:

from pyspark.sql import SparkSession, Row, DataFrame
import json
import pandas as pd
import sagemaker_pyspark
from pydeequ.checks import *
from pydeequ.verification import *
import pydeequ

classpath = ":".join(sagemaker_pyspark.classpath_jars())

df = spark.read.parquet("s3a://amazon-reviews-pds/parquet/product_category=Electronics/")
df.printSchema()

spark = (SparkSession
.builder
.config("spark.driver.extraClassPath", classpath)
.config("spark.jars.packages", pydeequ.deequ_maven_coord)
.config("spark.jars.excludes", pydeequ.f2j_maven_coord)
.getOrCreate())

check = Check(spark, CheckLevel.Warning, "Amazon Electronic Products Reviews")

checkResult = VerificationSuite(spark) \
.onData(df) \
.addCheck(
check.hasSize(lambda x: x >= 3000000) \
.hasMin("star_rating", lambda x: x == 1.0) \
.hasMax("star_rating", lambda x: x == 5.0)) \
.run()

print(f"Verification Run Status: {checkResult.status}")
checkResult_df = VerificationResult.checkResultsAsDataFrame(spark, checkResult, pandas=True)
checkResult_df

**Expected behavior**
The following table should get all success values.
![image](https://user-images.githubusercontent.com/76601107/125368581-7ed57180-e348-11eb-9de9-cb909b648516.png)

**Screenshots**
![image](https://user-images.githubusercontent.com/76601107/125369346-096aa080-e34a-11eb-9bb8-201cccc134f9.png)

The major issue is the value in the dataframe "Can't execute the assertion: An exception was raised by the Python Proxy. Return Message: null! "

Contributor guide

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.