def test_gibbs_smoke(): """Check if we don't get NaNs sampling the full digits dataset.""" rng = np.random.RandomState(42) X = Xdigits.astype(np.float32) rbm1 = BernoulliRBM(X.shape[1], n_hidden=42, batch_size=40, n_iter=20, random_state=rng) rbm1.fit(X) X_sampled = rbm1.gibbs(X) assert_all_finite(X_sampled)