No module named 'torch._subclasses.schema_check_mode'

[zhoushican]

LDA_obj.plot_topic_contributions(6)

When I run this line of code, I get the following error:
No module named 'torch._subclasses.schema_check_mode'

[HendricksJudy]

Sir, After testing the script

No, running LDA_obj •plot_topic_contributions (6) still results in the error 'No module named
torch._subclasses.schema_check_mode'.

The error is likely due to the absence of the torch._subclasses. schema_check_mode module in the current codebase.
The omicverse/bulk2single/_bulk2single.py file imports torch, but does not include any reference to
torch._subclasses.schema_check_mode .

The omicverse/bulk2single/_bulktrajblend.py file also imports torch, but does not include any reference to
torch._subclasses. schema_check_mode.

The omicverse/utils/_cluster.py file contains the LDA_topic class with the plot_topic_contributions method, but
does not include any reference to torch._subclasses.schema_check_mode .

You may check your PyTorch integrity and the version to solve this error