Fixing lint problems

dmlc · Feb 5, 2024 · 7f46f22 · 7f46f22
1 parent 982537c
commit 7f46f22
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tests/python/pytorch/graphbolt/impl/test_ondisk_dataset.py b/tests/python/pytorch/graphbolt/impl/test_ondisk_dataset.py
@@ -1928,7 +1928,7 @@ def test_OnDiskDataset_preprocess_force_preprocess(capsys):
                 gb.ondisk_dataset.preprocess_ondisk_dataset(
                     test_dir,
                     include_original_edge_id=False,
-                    force_preprocess=False
+                    force_preprocess=False,
                 )
             )
         captured = capsys.readouterr().out.split("\n")
@@ -1967,7 +1967,7 @@ def test_OnDiskDataset_preprocess_force_preprocess(capsys):
                 gb.ondisk_dataset.preprocess_ondisk_dataset(
                     test_dir,
                     include_original_edge_id=False,
-                    force_preprocess=True
+                    force_preprocess=True,
                 )
             )
         captured = capsys.readouterr().out.split("\n")
@@ -2974,7 +2974,7 @@ def test_OnDiskDataset_auto_force_preprocess(capsys):
         with pytest.warns(
             DGLWarning,
             match="Edge feature is stored, but edge IDs are not saved.",
-         ):
+        ):
             dataset = gb.OnDiskDataset(
                 test_dir, include_original_edge_id=False
             ).load()