rename folder tgb_modules to modules for merge into tgb

shenyangHuang · Jun 21, 2024 · 8cc0539 · 8cc0539
1 parent c2fefde
commit 8cc0539
Show file tree

Hide file tree

Showing 26 changed files with 6 additions and 6 deletions.
diff --git a/tgb_modules/decoder.py → modules/decoder.py b/tgb_modules/decoder.py → modules/decoder.py
diff --git a/tgb_modules/early_stopping.py → modules/early_stopping.py b/tgb_modules/early_stopping.py → modules/early_stopping.py
diff --git a/tgb_modules/edgebank_predictor.py → modules/edgebank_predictor.py b/tgb_modules/edgebank_predictor.py → modules/edgebank_predictor.py
diff --git a/tgb_modules/emb_module.py → modules/emb_module.py b/tgb_modules/emb_module.py → modules/emb_module.py
diff --git a/tgb_modules/heuristics.py → modules/heuristics.py b/tgb_modules/heuristics.py → modules/heuristics.py
diff --git a/tgb_modules/memory_module.py → modules/memory_module.py b/tgb_modules/memory_module.py → modules/memory_module.py
@@ -16,7 +16,7 @@
 from torch_geometric.nn.inits import zeros
 from torch_geometric.utils import scatter
 
-from tgb_modules.time_enc import TimeEncoder
+from modules.time_enc import TimeEncoder
 
 
 TGNMessageStoreType = Dict[int, Tuple[Tensor, Tensor, Tensor, Tensor]]

diff --git a/tgb_modules/msg_agg.py → modules/msg_agg.py b/tgb_modules/msg_agg.py → modules/msg_agg.py
diff --git a/tgb_modules/msg_func.py → modules/msg_func.py b/tgb_modules/msg_func.py → modules/msg_func.py
diff --git a/tgb_modules/neighbor_loader.py → modules/neighbor_loader.py b/tgb_modules/neighbor_loader.py → modules/neighbor_loader.py
diff --git a/tgb_modules/nodebank.py → modules/nodebank.py b/tgb_modules/nodebank.py → modules/nodebank.py
diff --git a/tgb_modules/recurrencybaseline_predictor.py → modules/recurrencybaseline_predictor.py b/tgb_modules/recurrencybaseline_predictor.py → modules/recurrencybaseline_predictor.py
@@ -14,7 +14,7 @@
 import numpy as np
 from collections import Counter
 import ray
-from tgb_modules.tkg_utils import create_scores_array
+from modules.tkg_utils import create_scores_array
 
 @ray.remote
 def baseline_predict_remote(num_queries, test_data, all_data, window, basis_dict, num_nodes, 

diff --git a/tgb_modules/rgcn_layers.py → modules/rgcn_layers.py b/tgb_modules/rgcn_layers.py → modules/rgcn_layers.py
diff --git a/tgb_modules/rgcn_model.py → modules/rgcn_model.py b/tgb_modules/rgcn_model.py → modules/rgcn_model.py
diff --git a/tgb_modules/rrgcn.py → modules/rrgcn.py b/tgb_modules/rrgcn.py → modules/rrgcn.py
@@ -6,9 +6,9 @@
 import torch.nn.functional as F
 import math
 
-from tgb_modules.rgcn_layers import UnionRGCNLayer, RGCNBlockLayer
-from tgb_modules.rgcn_model import BaseRGCN
-from tgb_modules.decoder import ConvTransE
+from modules.rgcn_layers import UnionRGCNLayer, RGCNBlockLayer
+from modules.rgcn_model import BaseRGCN
+from modules.decoder import ConvTransE
 import numpy as np
 class RGCNCell(BaseRGCN):
     def build_hidden_layer(self, idx):

diff --git a/tgb_modules/sthn.py → modules/sthn.py b/tgb_modules/sthn.py → modules/sthn.py
diff --git a/tgb_modules/time_enc.py → modules/time_enc.py b/tgb_modules/time_enc.py → modules/time_enc.py
diff --git a/tgb_modules/timetraveler_agent.py → modules/timetraveler_agent.py b/tgb_modules/timetraveler_agent.py → modules/timetraveler_agent.py
diff --git a/tgb_modules/timetraveler_dirichlet.py → modules/timetraveler_dirichlet.py b/tgb_modules/timetraveler_dirichlet.py → modules/timetraveler_dirichlet.py
diff --git a/tgb_modules/timetraveler_environment.py → modules/timetraveler_environment.py b/tgb_modules/timetraveler_environment.py → modules/timetraveler_environment.py
diff --git a/tgb_modules/timetraveler_episode.py → modules/timetraveler_episode.py b/tgb_modules/timetraveler_episode.py → modules/timetraveler_episode.py
diff --git a/tgb_modules/timetraveler_policygradient.py → modules/timetraveler_policygradient.py b/tgb_modules/timetraveler_policygradient.py → modules/timetraveler_policygradient.py
diff --git a/tgb_modules/timetraveler_trainertester.py → modules/timetraveler_trainertester.py b/tgb_modules/timetraveler_trainertester.py → modules/timetraveler_trainertester.py
diff --git a/tgb_modules/tkg_utils.py → modules/tkg_utils.py b/tgb_modules/tkg_utils.py → modules/tkg_utils.py
diff --git a/tgb_modules/tkg_utils_dgl.py → modules/tkg_utils_dgl.py b/tgb_modules/tkg_utils_dgl.py → modules/tkg_utils_dgl.py
diff --git a/tgb_modules/tlogic_apply_modules.py → modules/tlogic_apply_modules.py b/tgb_modules/tlogic_apply_modules.py → modules/tlogic_apply_modules.py
@@ -8,7 +8,7 @@
 import numpy as np
 import pandas as pd
 
-from tgb_modules.tlogic_learn_modules import store_edges
+from modules.tlogic_learn_modules import store_edges
 
 
 def filter_rules(rules_dict, min_conf, min_body_supp, rule_lengths):

diff --git a/tgb_modules/tlogic_learn_modules.py → modules/tlogic_learn_modules.py b/tgb_modules/tlogic_learn_modules.py → modules/tlogic_learn_modules.py