diff --git a/nobrainer/processing/base.py b/nobrainer/processing/base.py
index e7445674..f4df69fe 100644
--- a/nobrainer/processing/base.py
+++ b/nobrainer/processing/base.py
@@ -55,11 +55,11 @@ def save(self, save_dir):
 
     @classmethod
     def load(
-            cls,
-            model_dir,
-            multi_gpu=True,
-            custom_objects=None,
-            compile=False,
+        cls,
+        model_dir,
+        multi_gpu=True,
+        custom_objects=None,
+        compile=False,
     ):
         """Loads a trained model from a save directory"""
         model_dir = Path(str(model_dir).rstrip(os.pathsep))
@@ -87,13 +87,13 @@ def load(
 
     @classmethod
     def init_with_checkpoints(
-            cls,
-            model_name,
-            checkpoint_filepath,
-            multi_gpu=True,
-            custom_objects=None,
-            compile=False,
-            model_args=None,
+        cls,
+        model_name,
+        checkpoint_filepath,
+        multi_gpu=True,
+        custom_objects=None,
+        compile=False,
+        model_args=None,
     ):
         """Initialize a model for training, either from the latest
         checkpoint found, or from scratch if no checkpoints are
diff --git a/nobrainer/processing/checkpoint.py b/nobrainer/processing/checkpoint.py
index e8409c2b..e9cb9585 100644
--- a/nobrainer/processing/checkpoint.py
+++ b/nobrainer/processing/checkpoint.py
@@ -36,11 +36,11 @@ def save(self, directory):
         self.estimator.save(directory)
 
     def load(
-            self,
-            multi_gpu=True,
-            custom_objects=None,
-            compile=False,
-            model_args=None,
+        self,
+        multi_gpu=True,
+        custom_objects=None,
+        compile=False,
+        model_args=None,
     ):
         """Loads the most-recently created checkpoint from the
         checkpoint directory.