Modified value for pyspark value on validation

canimus · Jul 14, 2024 · 5e2bbae · 5e2bbae
1 parent 0c11e78
commit 5e2bbae
Showing 1 changed file with 4 additions and 5 deletions.
diff --git a/cuallee/pyspark_validation.py b/cuallee/pyspark_validation.py
@@ -801,11 +801,10 @@ def summary(check: Check, dataframe: DataFrame) -> DataFrame:
         spark = SparkSession.builder.getOrCreate()
 
     def _value(x):
-        """Removes verbosity for Callable values"""                
-        if x.options and isinstance(x.options, dict):
-                return x.options.get("custom_value", "f(x)")        
-        elif isinstance(x.value, Callable):
-            return "f(x)"
+        """Removes verbosity for Callable values"""
+        if isinstance(x.value, Callable):
+            if x.options and isinstance(x.options, dict):
+                return x.options.get("custom_value", "f(x)")
         else:
             return str(x.value)