ho22joshua commited on
Commit
d1b7df5
·
1 Parent(s): 2185661

fixed imports

Browse files
root_gnn_dgl/scripts/inference.py CHANGED
@@ -1,5 +1,6 @@
1
  import sys
2
- file_path = "/global/cfs/projectdirs/atlas/joshua/root_gnn/root_gnn_dgl"
 
3
  sys.path.append(file_path)
4
  import os
5
  import argparse
 
1
  import sys
2
+ import os
3
+ file_path = os.getcwd()
4
  sys.path.append(file_path)
5
  import os
6
  import argparse
root_gnn_dgl/scripts/prep_data.py CHANGED
@@ -1,5 +1,6 @@
1
  import sys
2
- file_path = "/global/cfs/projectdirs/atlas/joshua/root_gnn/root_gnn_dgl"
 
3
  sys.path.append(file_path)
4
 
5
  import root_gnn_base.utils as utils
 
1
  import sys
2
+ import os
3
+ file_path = os.getcwd()
4
  sys.path.append(file_path)
5
 
6
  import root_gnn_base.utils as utils
root_gnn_dgl/scripts/training_script.py CHANGED
@@ -11,7 +11,7 @@ import torch
11
  import torch.nn as nn
12
 
13
  import sys
14
- file_path = "/global/cfs/projectdirs/atlas/joshua/root_gnn/root_gnn_dgl/"
15
  sys.path.append(file_path)
16
  import root_gnn_base.batched_dataset as datasets
17
  from root_gnn_base import utils
 
11
  import torch.nn as nn
12
 
13
  import sys
14
+ file_path = os.getcwd()
15
  sys.path.append(file_path)
16
  import root_gnn_base.batched_dataset as datasets
17
  from root_gnn_base import utils