Format code with black

jameschapman19 · Oct 6, 2023 · 7d1950e · 7d1950e
1 parent a969fce
commit 7d1950e
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/docs/source/examples/plot_gradient.py b/docs/source/examples/plot_gradient.py
@@ -69,9 +69,7 @@
 score_display = ScoreScatterDisplay.from_estimator(
     cca, [X_train, Y_train], [X_test, Y_test]
 )
-score_display.plot(
-    title=f"CCA (Time: {elapsed_time:.2f} s)"
-)
+score_display.plot(title=f"CCA (Time: {elapsed_time:.2f} s)")
 plt.show()
 
 # %%
@@ -102,7 +100,9 @@
 
     # We plot the transformed views on a scatter plot with different colors for train and test sets
     # Use ScoreScatterDisplay or a similar plotting class for the visualization
-    score_display = ScoreScatterDisplay.from_estimator(ccaey, [X_train, Y_train], [X_test, Y_test])
+    score_display = ScoreScatterDisplay.from_estimator(
+        ccaey, [X_train, Y_train], [X_test, Y_test]
+    )
     score_display.plot(
         title=f"CCA_EY (Batch size: {batch_size}, Time: {elapsed_time:.2f} s)"
     )