James McCool commited on
Commit
229b9de
·
1 Parent(s): e8c9dac

Update data type handling for salary, median, and ownership difference calculations in reassess_edge.py to improve memory efficiency and precision.

Browse files
Files changed (1) hide show
  1. global_func/reassess_edge.py +3 -3
global_func/reassess_edge.py CHANGED
@@ -58,9 +58,9 @@ def reassess_edge(refactored_frame: pd.DataFrame, original_frame: pd.DataFrame,
58
  refactored_df = refactored_frame.copy()
59
  refactored_df = refactored_df.reset_index(drop=True)
60
 
61
- refactored_df['salary_diff'] = refactored_df['salary'] - orig_df['salary']
62
- refactored_df['median_diff'] = refactored_df['median'] - orig_df['median']
63
- refactored_df['own_diff'] = refactored_df['Own'] - orig_df['Own']
64
 
65
  change_mask = refactored_df[refactored_df['median_diff'] != 0]
66
 
 
58
  refactored_df = refactored_frame.copy()
59
  refactored_df = refactored_df.reset_index(drop=True)
60
 
61
+ refactored_df['salary_diff'] = (refactored_df['salary'] - orig_df['salary']).astype('int16')
62
+ refactored_df['median_diff'] = (refactored_df['median'] - orig_df['median']).astype('float32')
63
+ refactored_df['own_diff'] = (refactored_df['Own'] - orig_df['Own']).astype('float32')
64
 
65
  change_mask = refactored_df[refactored_df['median_diff'] != 0]
66