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

Fixed bug in removing output file.

Tom Flucke преди 6 години
родител
ревизия
05f82d2c6a
променени са 1 файла, в които са добавени 2 реда и са изтрити 1 реда
  1. 2 1
      src/feature-extractor/extractor.py

+ 2 - 1
src/feature-extractor/extractor.py

@@ -85,7 +85,8 @@ def main():
     import numpy as np
     if len(np.unique([s.user for s in out])) < args.min_users:
         print("Not enough data to build valid dataset.", file=sys.stderr)
-        os.remove(args.outfile.name)
+        if os.path.isfile(args.outfile.name):
+            os.remove(args.outfile.name)
         return
     try:
         import cPickle as pickle