@@ -395,7 +395,8 @@ class MigrationCleaning:
data[column] = data[column].astype(python_type)
elif python_type == float:
- data = data.fillna(np.inf)
+
+ data[column] = data[column].fillna(np.inf)
else: