Skip to content

Commit 60964fd

Browse files
committed
Fix stale _graphdef import paths after subpackage rename
Update two references that still used _graphdef instead of _graph_def after the subpackage split. Made-with: Cursor
1 parent 17715f2 commit 60964fd

2 files changed

Lines changed: 3 additions & 3 deletions

File tree

cuda_core/cuda/core/_graph/_graph_builder.pyx

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -793,12 +793,12 @@ class Graph:
793793

794794
Parameters
795795
----------
796-
source : :obj:`~_graph.GraphBuilder` or :obj:`~_graph._graphdef.GraphDef`
796+
source : :obj:`~_graph.GraphBuilder` or :obj:`~_graph._graph_def.GraphDef`
797797
The graph definition to update from. A GraphBuilder must have
798798
finished building.
799799

800800
"""
801-
from cuda.core._graph._graphdef import GraphDef
801+
from cuda.core._graph._graph_def import GraphDef
802802

803803
cdef cydriver.CUgraph cu_graph
804804
cdef cydriver.CUgraphExec cu_exec = <cydriver.CUgraphExec><intptr_t>int(self._mnff.graph)

cuda_core/tests/graph/test_graph_update.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@
88
from helpers.graph_kernels import compile_common_kernels, compile_conditional_kernels
99

1010
from cuda.core import Device, LaunchConfig, LegacyPinnedMemoryResource, launch
11-
from cuda.core._graph._graphdef import GraphDef
11+
from cuda.core._graph._graph_def import GraphDef
1212
from cuda.core._utils.cuda_utils import CUDAError
1313

1414

0 commit comments

Comments
 (0)