diff --git a/agml/data/loader.py b/agml/data/loader.py
index b0143acd..bbf6f8d6 100644
--- a/agml/data/loader.py
+++ b/agml/data/loader.py
@@ -23,24 +23,14 @@
 
 import numpy as np
 
-from agml.framework import AgMLSerializable
-from agml.data.manager import DataManager
-from agml.data.builder import DataBuilder
-from agml.data.metadata import DatasetMetadata, make_metadata
-from agml.data.exporters.yolo import export_yolo
-from agml.utils.logging import log
-from agml.utils.io import get_file_list, get_dir_list
-from agml.utils.data import load_public_sources
-from agml.utils.general import NoArgument, resolve_list_value
-from agml.utils.random import inject_random_state
-from agml.backend.config import data_save_path, synthetic_data_save_path, SUPER_BASE_DIR
+from agml.backend.config import SUPER_BASE_DIR, data_save_path, synthetic_data_save_path
 from agml.backend.experimental import AgMLExperimentalFeatureWrapper
 from agml.backend.tftorch import (
+    StrictBackendError,
+    _add_dataset_to_mro,  # noqa
     get_backend,
     set_backend,
     user_changed_backend,
-    StrictBackendError,
-    _add_dataset_to_mro,  # noqa
 )
 from agml.data.builder import DataBuilder
 from agml.data.exporters.yolo import export_yolo
diff --git a/agml/synthetic/config.py b/agml/synthetic/config.py
index 1c097bb1..33aa4413 100644
--- a/agml/synthetic/config.py
+++ b/agml/synthetic/config.py
@@ -28,13 +28,11 @@
 _HELIOS_CHECK_DONE_IN_SESSION = False
 
 # Paths to the Helios module and the relevant C++ files.
-HELIOS_PATH = os.path.join(recursive_dirname(__file__, 2), '_helios/Helios')
-PROJECT_ROOT = os.path.join(HELIOS_PATH, 'projects')
-CANOPY_SOURCE = os.path.join(
-    HELIOS_PATH, 'plugins/canopygenerator/src/CanopyGenerator.cpp')
-CANOPY_HEADER = os.path.join(
-    HELIOS_PATH, 'plugins/canopygenerator/include/CanopyGenerator.h')
-LIDAR_SOURCE = os.path.join(HELIOS_PATH, 'plugins/lidar/src/LiDAR.cpp')
+HELIOS_PATH = os.path.join(recursive_dirname(__file__, 2), "_helios/Helios")
+PROJECT_ROOT = os.path.join(HELIOS_PATH, "projects")
+CANOPY_SOURCE = os.path.join(HELIOS_PATH, "plugins/canopygenerator/src/CanopyGenerator.cpp")
+CANOPY_HEADER = os.path.join(HELIOS_PATH, "plugins/canopygenerator/include/CanopyGenerator.h")
+LIDAR_SOURCE = os.path.join(HELIOS_PATH, "plugins/lidar/src/LiDAR.cpp")
 
 # Path to the stored Helios configuration JSON file in root dir.
 HELIOS_CONFIG_FILE = os.path.expanduser("~/.agml/helios_config.json")
diff --git a/scripts/docs/gen_dataset_list.py b/scripts/docs/gen_dataset_list.py
index 7ee76830..97868136 100644
--- a/scripts/docs/gen_dataset_list.py
+++ b/scripts/docs/gen_dataset_list.py
@@ -1,10 +1,10 @@
 import json
-from textwrap import dedent
 from importlib import resources
+from textwrap import dedent
 
-
-from jinja2.sandbox import SandboxedEnvironment
 from jinja2 import StrictUndefined
+from jinja2.sandbox import SandboxedEnvironment
+
 # Check whether any dataset has been added that needs information updated (we do this
 # by checking for any datasets in the `public_datasources.json` file that don't have
 # corresponding information for the dataset statistics or other properties like shape).
diff --git a/tests/config/test_config.py b/tests/config/test_config.py
index 2598e61a..60913d5b 100644
--- a/tests/config/test_config.py
+++ b/tests/config/test_config.py
@@ -22,5 +22,3 @@ def test_config_setup():
     assert cfg.data_save_path() == cfg._get_config("data_path")
     assert cfg.model_save_path() == cfg._get_config("model_path")
     assert cfg.synthetic_data_save_path() == cfg._get_config("synthetic_data_path")
-
-    
diff --git a/tests/conftest.py b/tests/conftest.py
index ca6a518a..7263059d 100644
--- a/tests/conftest.py
+++ b/tests/conftest.py
@@ -1,9 +1,8 @@
 import os
-from pathlib import Path
 import shutil
+from pathlib import Path
 
 import pytest
-
 from pytest import TempPathFactory