fix actions

Author: SkafteNicki

s2_organisation_and_version_control/exercise_files/train_solution.py

@@ -1,9 +1,9 @@
 import matplotlib.pyplot as plt
 import torch
 import typer
-from model import MyAwesomeModel
+from model import MyAwesomeModel  # noqa: I001
 
-from data import corrupt_mnist
+from data import corrupt_mnist  # noqa: I001
 
 DEVICE = torch.device("cuda" if torch.cuda.is_available() else "mps" if torch.backends.mps.is_available() else "cpu")