diff --git a/setup.cfg b/setup.cfg
index 2d6e73e..267dfa0 100644
--- a/setup.cfg
+++ b/setup.cfg
@@ -23,7 +23,6 @@ install_requires =
     csbdeep
     lightning
     networkx
-    numerize
     numpy<2.0.0
     pandas
     Pillow
diff --git a/tests/test_training.py b/tests/test_training.py
index 6a91413..4e7ab42 100644
--- a/tests/test_training.py
+++ b/tests/test_training.py
@@ -1,7 +1,6 @@
 from spotiflow.data import SpotsDataset
 from spotiflow.model import SpotiflowModelConfig, SpotiflowTrainingConfig, Spotiflow
 import lightning.pytorch as pl
-from numerize.numerize import numerize
 import torch
 from utils import example_data
 
@@ -54,7 +53,7 @@
 
     model = Spotiflow(config)
 
-    print(f"Total params: {numerize(sum(p.numel() for p in model.parameters()))}")
+    print(f"Total params: {sum(p.numel() for p in model.parameters())}")
 
     logger = pl.loggers.TensorBoardLogger(
         save_dir="foo",
diff --git a/tests/test_training_simple.py b/tests/test_training_simple.py
index 1aa2d90..4617cb4 100644
--- a/tests/test_training_simple.py
+++ b/tests/test_training_simple.py
@@ -1,5 +1,4 @@
 from spotiflow.model import SpotiflowModelConfig, Spotiflow
-from numerize.numerize import numerize
 import torch
 from utils import example_data
 
@@ -15,6 +14,6 @@
 
     model = Spotiflow(config)
 
-    print(f"Total params: {numerize(sum(p.numel() for p in model.parameters()))}")
+    print(f"Total params: {sum(p.numel() for p in model.parameters())}")
 
     model.fit(X, P, Xv, Pv, save_dir="tmp", train_config={"num_epochs": 10})