Error when executing main.py

[agoodweathercc]

Hello,

When I execute main.py. I had the following error.

Traceback (most recent call last):
File "main.py", line 1, in
from train import *
File "/home/Documents/DeepLearning/graph-generation/train.py", line 24, in
from utils import *
File "/home/Documents/DeepLearning/graph-generation/utils.py", line 17, in
import data
File "/home/Documents/DeepLearning/graph-generation/data.py", line 17, in
from model import *
File "/home/Documents/DeepLearning/graph-generation/model.py", line 989
prob = x_prev @ x_last.permute(0,2,1)
^
SyntaxError: invalid syntax

[Takonan]

Are you using Python 3? The @ operator performs matmul: https://docs.scipy.org/doc/numpy/reference/generated/numpy.matmul.html#numpy.matmul