Преглед на файлове

update convert-types in mirgartoin cleaning

tsteuer преди 4 години
родител
ревизия
5f91f99edd
променени са 1 файла, в които са добавени 2 реда и са изтрити 1 реда
  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: