diff --git a/dacapo/train.py b/dacapo/train.py
index c940b8889..6f8894d7b 100644
--- a/dacapo/train.py
+++ b/dacapo/train.py
@@ -12,7 +12,7 @@
 logger = logging.getLogger(__name__)
 
 
-def train(run_name: str, compute_context: ComputeContext = LocalTorch(), force_cuda = False):
+def train(run_name: str, compute_context: ComputeContext = LocalTorch()):
     """Train a run"""
 
     if compute_context.train(run_name):