diff --git a/python/cuml/cuml/manifold/t_sne.pyx b/python/cuml/cuml/manifold/t_sne.pyx index 01ea7c0957..0d047c59d0 100644 --- a/python/cuml/cuml/manifold/t_sne.pyx +++ b/python/cuml/cuml/manifold/t_sne.pyx @@ -727,4 +727,4 @@ class TSNE(UniversalBase, def get_attr_names(self): return ["embedding", "kl_divergence_", "n_features_in_", "learning_rate_", - "n_iter_"] + "n_iter_", "embedding_"]