Bläddra i källkod

update convert-types in mirgartoin cleaning

tsteuer 4 år sedan
förälder
incheckning
5f91f99edd
1 ändrade filer med 2 tillägg och 1 borttagningar
  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: