Commit 242ff9b9 authored by evanfeinberg's avatar evanfeinberg
Browse files

pylint/nosestests fixes

parent 03a118b7
Loading
Loading
Loading
Loading
+1 −2
Original line number Diff line number Diff line
@@ -161,8 +161,7 @@ class Model(object):
      y_preds = []
      for j in range(len(interval_points)-1):
        indices = range(interval_points[j], interval_points[j+1])
        y_pred_on_batch = self.predict_on_batch(X[indices, :])
        #y_pred_on_batch = np.reshape(y_pred_on_batch, (len(indices),))
        y_pred_on_batch = self.predict_on_batch(X[indices, :]).reshape((len(indices),len(task_names)))
        y_preds.append(y_pred_on_batch)

      y_pred = np.concatenate(y_preds)
+4 −6
Original line number Diff line number Diff line
@@ -191,8 +191,6 @@ class DataFeaturizer(object):
  def _standardize_df(self, ori_df):
    """Copy specified columns to new df with standard column names."""
    df = pd.DataFrame([])
    print("ori_df.keys()")
    print(ori_df.keys())
    df["mol_id"] = ori_df[[self.id_field]]
    df["smiles"] = ori_df[[self.smiles_field]]
    for task in self.tasks:
+1 −1

File changed.

Contains only whitespace changes.