Commit 31b211a0 authored by Vignesh's avatar Vignesh
Browse files

Fix data-loader failing tests

parent 7af7edae
Loading
Loading
Loading
Loading
+2 −1
Original line number Diff line number Diff line
@@ -128,7 +128,8 @@ def get_user_specified_features(df, featurizer, verbose=True):
  time1 = time.time()
  df[featurizer.feature_fields] = df[featurizer.feature_fields].apply(
      pd.to_numeric)
  X_shard = df.as_matrix(columns=featurizer.feature_fields)
  df_fields = df[featurizer.feature_fields]
  X_shard = df_fields.values
  time2 = time.time()
  log("TIMING: user specified processing took %0.3f s" % (time2 - time1),
      verbose)