Skip to content

Commit

Permalink
fix: iter & num_iter not needed for the preprocessing function
Browse files Browse the repository at this point in the history
  • Loading branch information
yiitozer committed Mar 20, 2024
1 parent 924ec50 commit edf30ee
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion libnmfd/core/nmfconv.py
Original file line number Diff line number Diff line change
Expand Up @@ -274,7 +274,7 @@ def nmfd(V: np.ndarray,
for iter in tnrange(L, desc='Processing'):
# if given from the outside, apply soft constraints
if func_preprocess is not None:
tensor_W, H = func_preprocess(tensor_W, H, iter, L, **kwargs)
tensor_W, H = func_preprocess(W=tensor_W, H=H, **kwargs)

# compute first approximation
Lambda = conv_model(tensor_W, H)
Expand Down

0 comments on commit edf30ee

Please sign in to comment.