Skip to content
Merged
Changes from 2 commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
3 changes: 1 addition & 2 deletions src/transformers/models/graphormer/collating_graphormer.py
Original file line number Diff line number Diff line change
Expand Up @@ -6,7 +6,7 @@
import numpy as np
import torch

from ...utils import is_cython_available, requires_backends
from ...utils import is_cython_available


if is_cython_available():
Expand All @@ -24,7 +24,6 @@ def convert_to_single_emb(x, offset: int = 512):


def preprocess_item(item, keep_features=True):
requires_backends(preprocess_item, ["Cython"])
Comment thread
clefourrier marked this conversation as resolved.
if not is_cython_available():
raise ImportError("Graphormer preprocessing needs Cython (pyximport)")

Expand Down