def test_to_pickle(entityset): dirname = os.path.dirname(integration_data.__file__) path = os.path.join(dirname, 'test_entityset.p') if os.path.exists(path): shutil.rmtree(path) entityset.to_pickle(path) new_es = ft.read_pickle(path) assert entityset.__eq__(new_es, deep=True) shutil.rmtree(path)
encodeOutput = args.encodeOutput if not encodeOutput: sys.exit("Missing required argument: encodeOutput") # Assign WorkingDir to Feature Tools os.environ['FEATURETOOLS_DIR'] = workingDir # Import featuretools now that we've correctly set the env var. import featuretools as ft from featuretools.selection import remove_low_information_features # Unzip and Unpickle Input dir_name = workingDir + "output.pkl" shutil.unpack_archive(inFile.name, dir_name, 'zip') es = ft.read_pickle(dir_name, load_data=True) if not es: sys.exit("Missing required argument: entity set") # Adjust Entity Set cutoff_times = es['transactions'].df[['Transaction Id', 'End Time', 'Outcome']] pd.options.display.max_columns = 500 fm, features = ft.dfs(entityset=es, target_entity='transactions', agg_primitives=aggPrimitives, trans_primitives=transPrimitives, max_depth=maxDepth, cutoff_time=cutoff_times[1000:],