Transcript_id overflows, must be torch.long

EliHei2 · Oct 16, 2024 · 4d3e4e3 · 4d3e4e3
1 parent ceeba81
commit 4d3e4e3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/segger/data/parquet/sample.py b/src/segger/data/parquet/sample.py
@@ -1127,7 +1127,7 @@ def to_pyg_dataset(
         # Set up Transcript nodes
         pyg_data["tx"].id = torch.tensor(
             self.transcripts[self.settings.transcripts.id].values.astype(int),
-            dtype=torch.int,
+            dtype=torch.long,
         )
         pyg_data["tx"].pos = torch.tensor(
             self.transcripts[self.settings.transcripts.xyz].values,