Commit 0a88267b authored by galenxing's avatar galenxing
Browse files

yapf

parent d662c00e
Loading
Loading
Loading
Loading
+2 −1
Original line number Diff line number Diff line
@@ -614,7 +614,8 @@ class GraphConvTensorGraph(TensorGraph):
        self.default_generator(dataset, predict=True),
        metrics,
        labels=self.my_labels,
        weights=[self.my_task_weights], per_task_metrics = per_task_metrics)
        weights=[self.my_task_weights],
        per_task_metrics=per_task_metrics)

  def bayesian_predict(self,
                       dataset,