Reformat code for changes related to Issue intel#651

aadarshsingh191198 · Oct 7, 2020 · 469ebd9 · 469ebd9
1 parent 0224747
commit 469ebd9
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/model/scikit/dffml_model_scikit/scikit_base.py b/model/scikit/dffml_model_scikit/scikit_base.py
@@ -148,7 +148,9 @@ async def predict(
             )
             target = self.parent.config.predict.name
             record.predicted(
-                target, self.parent.config.predict.dtype(self.clf.predict(predict)[0]), self.confidence
+                target,
+                self.parent.config.predict.dtype(self.clf.predict(predict)[0]),
+                self.confidence,
             )
             yield record
 
@@ -260,7 +262,11 @@ def yield_labels():
                 "Predicted cluster for {}: {}".format(predict, prediction)
             )
             target = self.parent.config.predict.name
-            record.predicted(target, self.parent.config.predict.dtype(prediction[0]), self.confidence)
+            record.predicted(
+                target,
+                self.parent.config.predict.dtype(prediction[0]),
+                self.confidence,
+            )
             yield record