Skip certain cugraph-pyg tests when torch-sparse is not available (#…

…3970)

A follow-up PR to fix the test failure again on rockylinux (see [log](https://github.com/rapidsai/cugraph/actions/runs/6718664898/job/18258783035#step:7:9297)).

`import_optional` only catches `ModuleNotFoundError`, however, it threw an `OSError` on rockylinux. Let's just catch all exceptions during imports in this case.

Closes rapidsai/graph_dl#373

Authors:
  - Tingyu Wang (https://github.com/tingyu66)

Approvers:
  - Rick Ratzel (https://github.com/rlratzel)
  - Brad Rees (https://github.com/BradReesWork)

URL: #3970

python/cugraph-pyg/cugraph_pyg/tests/test_cugraph_loader.py

@@ -29,9 +29,15 @@
 
 torch = import_optional("torch")
 torch_geometric = import_optional("torch_geometric")
-torch_sparse = import_optional("torch_sparse")
 trim_to_layer = import_optional("torch_geometric.utils.trim_to_layer")
 
+try:
+    import torch_sparse  # noqa: F401
+
+    HAS_TORCH_SPARSE = True
+except:  # noqa: E722
+    HAS_TORCH_SPARSE = False
+
 
 @pytest.mark.skipif(isinstance(torch, MissingModule), reason="torch not available")
 def test_cugraph_loader_basic(karate_gnn):
@@ -201,9 +207,7 @@ def test_cugraph_loader_from_disk_subset():
 
 
 @pytest.mark.skipif(isinstance(torch, MissingModule), reason="torch not available")
-@pytest.mark.skipif(
-    isinstance(torch_sparse, MissingModule), reason="torch-sparse not available"
-)
+@pytest.mark.skipif(not HAS_TORCH_SPARSE, reason="torch-sparse not available")
 def test_cugraph_loader_from_disk_subset_csr():
     m = [2, 9, 99, 82, 11, 13]
     n = torch.arange(1, 1 + len(m), dtype=torch.int32)
@@ -336,9 +340,7 @@ def test_cugraph_loader_e2e_coo():
 
 
 @pytest.mark.skipif(isinstance(torch, MissingModule), reason="torch not available")
-@pytest.mark.skipif(
-    isinstance(torch_sparse, MissingModule), reason="torch-sparse not available"
-)
+@pytest.mark.skipif(not HAS_TORCH_SPARSE, reason="torch-sparse not available")
 @pytest.mark.parametrize("framework", ["pyg", "cugraph-ops"])
 def test_cugraph_loader_e2e_csc(framework):
     m = [2, 9, 99, 82, 9, 3, 18, 1, 12]