fix: setting with copy warning

greenhub-project · Aug 23, 2019 · d17e156 · d17e156
1 parent e047859
commit d17e156
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/app/app.py b/app/app.py
@@ -39,11 +39,11 @@ def downcast_df(df):
 
   # transform optimized types
   if converted_int is not None:
-    df[converted_int.columns] = converted_int
+    df.loc[:, converted_int.columns] = converted_int
   if converted_float is not None:
-    df[converted_float.columns] = converted_float
+    df.loc[:, converted_float.columns] = converted_float
   if converted_obj is not None:
-    df[converted_obj.columns] = converted_obj
+    df.loc[:, converted_obj.columns] = converted_obj
 
   return df
 

diff --git a/plugins/clean_samples.py b/plugins/clean_samples.py
@@ -12,7 +12,7 @@ def apply(df):
   # explicitly cast battery level to integer
   df_level = df.battery_level * 100
   converted_level = df_level.astype(np.uint8)
-  df['battery_level'] = converted_level
+  df.loc[:, 'battery_level'] = converted_level
 
   # filter out malformed records
   df = df[df.battery_level <= 100]