remove unnecessary dep

weigertlab · Jun 17, 2024 · dbeb880 · dbeb880
1 parent 84e9d74
commit dbeb880
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 5 deletions.
diff --git 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
@@ -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
@@ -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})