linting

Author: mfbalin

tests/python/pytorch/graphbolt/impl/test_neighbor_sampler.py

@@ -26,7 +26,9 @@ def get_hetero_graph(include_original_edge_ids):
         "mask": torch.BoolTensor([1, 0, 1, 0, 1, 1, 1, 0, 1, 1]),
     }
     if include_original_edge_ids:
-        edge_attributes[gb.ORIGINAL_EDGE_ID] = torch.arange(indices.size(0), 0, -1)
+        edge_attributes[gb.ORIGINAL_EDGE_ID] = torch.arange(
+            indices.size(0), 0, -1
+        )
     node_type_offset = torch.LongTensor([0, 1, 3, 6])
     return gb.fused_csc_sampling_graph(
         indptr,