From ec9867fea4c5330a9471d27f9d2f6d55ba68dc6b Mon Sep 17 00:00:00 2001 From: anpolol Date: Mon, 8 Jan 2024 14:16:32 +0300 Subject: [PATCH] mypy error fix --- stable_gnn/model_link_predict.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/stable_gnn/model_link_predict.py b/stable_gnn/model_link_predict.py index aa81be9..2f5180e 100644 --- a/stable_gnn/model_link_predict.py +++ b/stable_gnn/model_link_predict.py @@ -1,5 +1,5 @@ import random -from typing import List, Tuple +from typing import Any, List, Tuple import torch from sklearn.base import BaseEstimator @@ -38,7 +38,7 @@ def __init__( def train_test_edges( self, dataset: Graph - ) -> Tuple[List[List[int]], List[List[int]], List[List[int]], List[List[int]]]: + ) -> Tuple[List[Any], List[int], List[Any], List[int]]: """ Split dataset to train and test and calculate negative samples