Unverified Commit aff6b685 authored by Hongzhi (Steve), Chen's avatar Hongzhi (Steve), Chen Committed by GitHub
Browse files

[Misc] Avoid using Literal which requires python 3.8. (#6493)


Co-authored-by: default avatarUbuntu <ubuntu@ip-172-31-28-63.ap-northeast-1.compute.internal>
parent e6a15c1a
......@@ -39,8 +39,6 @@ main
"""
import argparse
from typing import Literal
import dgl.graphbolt as gb
import dgl.nn as dglnn
import torch
......@@ -50,9 +48,7 @@ import torchmetrics.functional as MF
from tqdm import tqdm
def create_dataloader(
args, graph, features, itemset, job: Literal["train", "evaluate", "infer"]
):
def create_dataloader(args, graph, features, itemset, job):
"""
[HIGHLIGHT]
Get a GraphBolt version of a dataloader for node classification tasks.
......@@ -70,7 +66,7 @@ def create_dataloader(
The node features.
itemset : Union[ItemSet, ItemSetDict]
Data to be sampled.
job : Literal["train", "evaluate", "infer"]
job : one of ["train", "evaluate", "infer"]
The stage where dataloader is created, with options "train", "evaluate"
and "infer".
"""
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment