AmirShabani commited on
Commit
366ed46
·
1 Parent(s): 5d32b92

Try pip out

Browse files
Files changed (1) hide show
  1. core.py +13 -4
core.py CHANGED
@@ -1,7 +1,16 @@
1
- import subprocess
2
- import sys
3
-
4
- subprocess.check_call([sys.executable, "-m", "pip", "install", "./post-requirements.txt"])
 
 
 
 
 
 
 
 
 
5
 
6
  import pickle
7
  import numpy as np
 
1
+ import pip
2
+
3
+ def install(package):
4
+ if hasattr(pip, 'main'):
5
+ pip.main(['install', package])
6
+ else:
7
+ pip._internal.main(['install', package])
8
+
9
+ print("Everything goes bang")
10
+ install('torch_geometric')
11
+ install('torch_scatter')
12
+ install('torch_sparse')
13
+ print("It's havoc baby")
14
 
15
  import pickle
16
  import numpy as np