Merge pull request #151 from PriorLabs/fix_fp_feature

fix: make fingerprint feature deterministic
PriorLabs · Jan 22, 2025 · aadaf75 · aadaf75
2 parents c895961 + 3f97bfe
commit aadaf75
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/tabpfn/model/preprocessing.py b/src/tabpfn/model/preprocessing.py
@@ -2,6 +2,7 @@
 
 from __future__ import annotations
 
+import hashlib
 import warnings
 from abc import abstractmethod
 from collections import UserList
@@ -474,8 +475,7 @@ def _transform(self, X: np.ndarray, *, is_test: bool = False) -> np.ndarray:
 
 
 def float_hash_arr(arr: np.ndarray) -> float:
-    b = arr.tobytes()
-    _hash = hash(b)
+    _hash = int(hashlib.sha256(arr.tobytes()).hexdigest(), 16)
     return _hash % _CONSTANT / _CONSTANT