CSDI.predict n_sampling_times is not used in func

[Durakavalyanie]

1. System Info

In CSDI predict method n_sampling_times arg. is not used and it always returns with one prediction.

2. Information

• The official example scripts
• My own created scripts

3. Reproduction

print(X_miss.shape)
X_miss = {"X": X_miss}
predict = csdi.predict(X_miss, n_sampling_times=999)
print(predict['imputation'].shape)

actual output:
(10, 100, 1)
(10, 1, 100, 1)

4. Expected behavior

expected output:
(10, 100, 1)
(10, 999, 100, 1)

5. Your contribution

PyPOTS/pypots/imputation/csdi/model.py

311. results = self.model(inputs)

to

311. results = self.model(inputs, n_sampling_times=n_sampling_times)

[github-actions]

Hi there 👋,

Thank you so much for your attention to PyPOTS! You can follow me on GitHub to receive the latest news of PyPOTS.
If you find PyPOTS helpful to your work, please star⭐️ this repository to get this issue prioritized.
Your star is your recognition, which can help more people notice PyPOTS and grow PyPOTS community.
It matters and is definitely a kind of contribution to the community.

Best,
Wenjie

[WenjieDu]

That's true. This bug could be a miss since the last refactoring was in a rush. Thank you for pointing it out, Pavel.

Your code correction is right. Would you be interested in making a PR to fix this? It could be your first contribution to PyPOTS ;-)

[github-actions]

It seems the issue was completed. Closing it now.