flake8/linter fixes

Author: brifordwylie

experiments/pytorch_model.py

@@ -556,5 +556,6 @@ def predict_fn(df, model) -> pd.DataFrame:
     # Remove the pytorch_outputs directory if it exists
     if os.path.exists("pytorch_outputs"):
         import shutil
+
         shutil.rmtree("pytorch_outputs")
-        print("Removed pytorch_outputs directory.")
+        print("Removed pytorch_outputs directory.")