|
1 |
| -# Copyright (c) 2022-2023, NVIDIA CORPORATION. |
| 1 | +# Copyright (c) 2022-2025, NVIDIA CORPORATION. |
2 | 2 | #
|
3 | 3 | # Licensed under the Apache License, Version 2.0 (the "License");
|
4 | 4 | # you may not use this file except in compliance with the License.
|
@@ -656,10 +656,12 @@ def test_backend_torch(
|
656 | 656 | def test_remote_graph_neighbor_sample(
|
657 | 657 | client_with_property_csvs_loaded, pG_with_property_csvs_loaded
|
658 | 658 | ):
|
659 |
| - # FIXME: consider moving the call dispatcher into cugraph-service-client or |
660 |
| - # cugraph proper. Import it here for now to allow tests to run in an |
661 |
| - # environment without cugraph-pyg. |
662 |
| - from cugraph_pyg.loader.dispatch import call_cugraph_algorithm |
| 659 | + # FIXME: This test was using call_cugraph_algorithm from |
| 660 | + # cugraph_pyg.loader.dispatch, which no longer exists. This test should be |
| 661 | + # updated to call the proper replacement for call_cugraph_algorithm, or |
| 662 | + # removed. |
| 663 | + def call_cugraph_algorithm(*args, **kwargs): |
| 664 | + raise NotImplementedError |
663 | 665 |
|
664 | 666 | rpG = RemoteGraph(client_with_property_csvs_loaded, 0)
|
665 | 667 | pG = pG_with_property_csvs_loaded
|
@@ -701,10 +703,12 @@ def test_remote_graph_neighbor_sample(
|
701 | 703 | def test_remote_graph_neighbor_sample_implicit_subgraph(
|
702 | 704 | client_with_property_csvs_loaded, pG_with_property_csvs_loaded
|
703 | 705 | ):
|
704 |
| - # FIXME: consider moving the call dispatcher into cugraph-service-client or |
705 |
| - # cugraph proper. Import it here for now to allow tests to run in an |
706 |
| - # environment without cugraph-pyg. |
707 |
| - from cugraph_pyg.loader.dispatch import call_cugraph_algorithm |
| 706 | + # FIXME: This test was using call_cugraph_algorithm from |
| 707 | + # cugraph_pyg.loader.dispatch, which no longer exists. This test should be |
| 708 | + # updated to call the proper replacement for call_cugraph_algorithm, or |
| 709 | + # removed. |
| 710 | + def call_cugraph_algorithm(*args, **kwargs): |
| 711 | + raise NotImplementedError |
708 | 712 |
|
709 | 713 | rpG = RemoteGraph(client_with_property_csvs_loaded, 0)
|
710 | 714 | pG = pG_with_property_csvs_loaded
|
|
0 commit comments