Ver código fonte

update convert-types in mirgartoin cleaning

tsteuer 4 anos atrás
pai
commit
5f91f99edd
1 arquivos alterados com 2 adições e 1 exclusões
  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: