소스 검색

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: