From 4b114ff3dfc2b9cb04b468aa6adf42e8acabffa4 Mon Sep 17 00:00:00 2001 From: Yozen Liu Date: Thu, 16 Oct 2025 23:52:04 -0700 Subject: [PATCH] cap pos --- python/gigl/utils/data_splitters.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/gigl/utils/data_splitters.py b/python/gigl/utils/data_splitters.py index a7272cd4d..259e0b047 100644 --- a/python/gigl/utils/data_splitters.py +++ b/python/gigl/utils/data_splitters.py @@ -684,7 +684,7 @@ def _get_padded_labels( ], dim=0, ) - return labels + return labels[:,:3] def _assert_sampling_direction(sampling_direction: str):