add

awslabs · Nov 4, 2024 · b7bcbaa · b7bcbaa
1 parent 1f085da
commit b7bcbaa
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/tests/unit-tests/gpartition/conftest.py b/tests/unit-tests/gpartition/conftest.py
@@ -21,8 +21,7 @@
 import pytest
 
 from graphstorm.gpartition import LocalPartitionAlgorithm
-from graphstorm.gpartition.post_hard_negative import (shuffle_hard_negative_nids,
-                                                      load_hard_negative_config)
+
 
 @pytest.fixture(scope="module", name="chunked_metadata_dict")
 def metadata_dict_fixture() -> Dict: