use float instead of int for default values

emdgroup · May 25, 2021 · ebe6d53 · ebe6d53
1 parent bfab65a
commit ebe6d53
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tnmf/TransformInvariantNMF.py b/tnmf/TransformInvariantNMF.py
@@ -153,7 +153,7 @@ def fit(
             V: np.ndarray,
             update_H: bool = True,
             update_W: bool = True,
-            sparsity_H: float = 0,
+            sparsity_H: float = 0.,
             inhibition_strength: float = 0.,
             progress_callback: Callable[['TransformInvariantNMF', int], bool] = None,
     ):
@@ -164,7 +164,7 @@ def partial_fit(
             V: np.ndarray,
             update_H: bool = True,
             update_W: bool = True,
-            sparsity_H: float = 0,
+            sparsity_H: float = 0.,
             inhibition_strength: float = 0.,
             progress_callback: Callable[['TransformInvariantNMF', int], bool] = None,
     ):