Skip to content

Commit 26db821

Browse files
authored
MNT add dtype preservation tag to BernoulliRBM (scikit-learn#24318)
1 parent fb4eb1e commit 26db821

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

sklearn/neural_network/_rbm.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -447,5 +447,6 @@ def _more_tags(self):
447447
"check_methods_sample_order_invariance": (
448448
"fails for the score_samples method"
449449
),
450-
}
450+
},
451+
"preserves_dtype": [np.float64, np.float32],
451452
}

0 commit comments

Comments
 (0)