better formatting

Author: alvaropp

beginner_source/basics/buildmodel_tutorial.py

@@ -36,7 +36,13 @@
 # if available. Let's check to see if `torch.cuda <https://pytorch.org/docs/stable/notes/cuda.html>`_
 # or `torch.backends.mps <https://pytorch.org/docs/stable/notes/mps.html>` are available, otherwise we use the CPU.
 
-device = "cuda" if torch.cuda.is_available() else "mps" if torch.backends.mps.is_available() else "cpu"
+device = (
+    "cuda"
+    if torch.cuda.is_available()
+    else "mps"
+    if torch.backends.mps.is_available()
+    else "cpu"
+)
 print(f"Using {device} device")
 
 ##############################################

beginner_source/basics/quickstart_tutorial.py

@@ -87,7 +87,13 @@
 # operations in the neural network, we move it to the GPU or MPS if available.
 
 # Get cpu, gpu or mps device for training.
-device = "cuda" if torch.cuda.is_available() else "mps" if torch.backends.mps.is_available() else "cpu"
+device = (
+    "cuda"
+    if torch.cuda.is_available()
+    else "mps"
+    if torch.backends.mps.is_available()
+    else "cpu"
+)
 print(f"Using {device} device")
 
 # Define model