Fix bug

sdv-dev · May 14, 2024 · b188f13 · b188f13
1 parent 0840c7f
commit b188f13
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/sdv/sequential/par.py b/sdv/sequential/par.py
@@ -178,7 +178,7 @@ def _transform_sequence_index(self, data):
             fill_value = min(sequence_index_sequence[self._sequence_index].dropna())
         sequence_index_sequence = sequence_index_sequence.fillna(fill_value)
 
-        data[self._sequence_index] = sequence_index_sequence[self._sequence_index]
+        data[self._sequence_index] = sequence_index_sequence[self._sequence_index].to_numpy()
         data = data.merge(
             sequence_index_context,
             left_on=self._sequence_key,