Update clean_csv.py

fix the try/except loop in importing datacleaner (fix test suite in docker).
jim-schwoebel · Aug 12, 2020 · 1517009 · 1517009
1 parent 9d7e859
commit 1517009
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/cleaning/csv_cleaning/clean_csv.py b/cleaning/csv_cleaning/clean_csv.py
@@ -40,6 +40,7 @@
     import datacleaner
 except:
     os.system('pip3 install datacleaner==0.1.5')
+    import datacleaner
 
 def clean_csv(csvfile, basedir):
     '''
@@ -49,4 +50,4 @@ def clean_csv(csvfile, basedir):
     newframe=datacleaner.autoclean(input_dataframe, drop_nans=False, copy=False, ignore_update_check=False)
     newfile='clean_'+csvfile
     newframe.to_csv(newfile, index=False)
-    return [newfile]
+    return [newfile]