Commit 0eaec579 authored by nd-02110114's avatar nd-02110114
Browse files

♻️ small refator

parent f41de198
Loading
Loading
Loading
Loading
+1 −1
Original line number Diff line number Diff line
@@ -52,7 +52,7 @@ def test_cgcnn():
  assert test_preds.shape == (10, n_tasks)

  # eval model on test
  regression_metric = Metric(mae_score, n_tasks=1)
  regression_metric = Metric(mae_score, n_tasks=n_tasks)
  scores = model.evaluate(test, [regression_metric])
  assert scores[regression_metric.name] < 1.0