Parcourir la source

update convert-types in mirgartoin cleaning

tsteuer il y a 4 ans
Parent
commit
5f91f99edd
1 fichiers modifiés avec 2 ajouts et 1 suppressions
  1. 2 1
      cdplib/db_migration/MigrationCleaning.py

+ 2 - 1
cdplib/db_migration/MigrationCleaning.py

@@ -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)
                     data[column] = data[column].astype(python_type)
 
                 else: