Handle round as dataframe operation

hi-primus · Oct 17, 2018 · c6a2b4f · c6a2b4f
1 parent aa32325
commit c6a2b4f
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/optimus/dataframe/columns.py b/optimus/dataframe/columns.py
@@ -447,9 +447,6 @@ def parse_col_names_funcs_to_keys(data):
                         temp_func_name = f + "_"
                         if k.startswith(temp_func_name):
                             _col_name = k[len(temp_func_name):]
-                            # If the value is numeric only get 5 decimals
-                            if is_numeric(v):
-                                v = round(v, 5)
                             _result.setdefault(_col_name, {})[f] = v
                 return _result
             else:
@@ -470,7 +467,8 @@ def parse_col_names_funcs_to_keys(data):
         expression = []
         for col_name in columns:
             for func in funcs:
-                expression.append(func(col_name).alias(func.__name__ + "_" + col_name))
+                # If the value is numeric only get 5 decimals
+                expression.append(F.round(func(col_name), 5).alias(func.__name__ + "_" + col_name))
 
         result = parse_col_names_funcs_to_keys(format_dict(df.agg(*expression).to_json()))
         # logging.info(result)