andreped commited on
Commit
98c3e5f
1 Parent(s): 0b72594

debugging setup

Browse files
Files changed (1) hide show
  1. setup.py +11 -6
setup.py CHANGED
@@ -1,6 +1,9 @@
1
  from setuptools import setup, find_packages
2
  from setuptools.command.install import install
3
- import os
 
 
 
4
 
5
 
6
  with open("README.md", "r") as f:
@@ -12,21 +15,23 @@ with open('requirements.txt', 'r', encoding='utf-16') as ff:
12
 
13
  class InstallCommand(install):
14
  user_option = install.user_options + [
15
- ('cupy=', 'cupy', 'enable flag to install package with GPU support'),
16
  ]
17
 
18
  def initialize_options(self):
19
- install.initialize_options(self)
20
  self.cupy = "cupy"
21
 
22
  def finalize_options(self):
23
- print("CuPy version selected is: ", self.cupy)
24
- install.finalize_options(self)
25
 
26
  def run(self):
 
27
  global cupy
28
  required.append(self.cupy)
29
- install.run(self)
 
30
 
31
 
32
  setup(
 
1
  from setuptools import setup, find_packages
2
  from setuptools.command.install import install
3
+ import os, sys
4
+
5
+
6
+ print("\n\n\n\n---:", sys.args)
7
 
8
 
9
  with open("README.md", "r") as f:
 
15
 
16
  class InstallCommand(install):
17
  user_option = install.user_options + [
18
+ ('cupyyy=', None, 'enable flag to install package with GPU support'),
19
  ]
20
 
21
  def initialize_options(self):
22
+ super().initialize_options()
23
  self.cupy = "cupy"
24
 
25
  def finalize_options(self):
26
+ print("CuPy version selected is: ", self.cupyyy)
27
+ super().finalize_options()
28
 
29
  def run(self):
30
+ # use options
31
  global cupy
32
  required.append(self.cupy)
33
+
34
+ super().run()
35
 
36
 
37
  setup(