error raised when denoising

[dhy2002]

when i run "adata = STAGATE.train_STAGATE(adata, alpha=0, save_reconstrction=True)"
it raises the
error: Value passed for key 'STAGATE_ReX' is of incorrect shape. Values of layers must match dimensions (0, 1) of parent. Value had shape (4481, 3000) while it should have had (4481, 36601).
so how can i solve this problem?
thanks very much.

[RucDongLab]

You can run "adata = adata[:, adata.var['highly_variable']]" before running the denosing task. This error raised because the dimension of the denoising matrix should be equal to the dimension of the SCANPY object.