fix try catch

alibaba · Jan 23, 2025 · bdcf38f · bdcf38f
1 parent 2c031d8
commit bdcf38f
Showing 1 changed file with 3 additions and 8 deletions.
diff --git a/tzrec/datasets/data_parser.py b/tzrec/datasets/data_parser.py
@@ -453,14 +453,9 @@ def _to_sparse_features(
                     seq_length = length
                     key_length = input_data[f"{key}.key_lengths"]
                     # TODO: remove to_float when segment_reduce support int values
-                    try:
-                        length = torch.segment_reduce(
-                            key_length.float(), "sum", lengths=seq_length
-                        ).to(length.dtype)
-                    except Exception:
-                        import pdb
-
-                        pdb.set_trace()
+                    length = torch.segment_reduce(
+                        key_length.float(), "sum", lengths=seq_length
+                    ).to(length.dtype)
                     mulval_keys.append(key)
                     mulval_seq_lengths.append(seq_length)
                     mulval_key_lengths.append(key_length)