|
|
@@ -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
|