fix

rapidsai · Mar 7, 2024 · a873b51 · a873b51
1 parent 9f4c461
commit a873b51
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/python/cugraph-dgl/cugraph_dgl/nn/conv/gatv2conv.py b/python/cugraph-dgl/cugraph_dgl/nn/conv/gatv2conv.py
@@ -11,9 +11,9 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from typing import Optional, Union
+from typing import Any, Optional, Union
 
-from cugraph_dgl.nn.conv.base import Any, BaseConv, SparseGraph
+from cugraph_dgl.nn.conv.base import BaseConv, SparseGraph
 from cugraph.utilities.utils import import_optional
 
 dgl = import_optional("dgl")