From bdce40718b5e1089c01b93be83f7bd0f72fde994 Mon Sep 17 00:00:00 2001
From: bingsanyu <12348372+bingsanyu@user.noreply.gitee.com~>
Date: Sun, 29 Oct 2023 23:10:44 +0800
Subject: [PATCH] Fixed the missing import statement.

---
 hgt.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/hgt.py b/hgt.py
index 3b8eb52..a722174 100755
--- a/hgt.py
+++ b/hgt.py
@@ -701,6 +701,7 @@ def add_budget(te, target_id, target_time, layer_data, budget):
         if (args.loss == "cross"):
             from torch_geometric.utils import remove_self_loops
             from torch_geometric.utils import add_self_loops
+            from torch_geometric.utils import negative_sampling
 
             EPS = 1e-15
             value = (node_rep[edge_index[0]] *