Skip to content

Commit

Permalink
added Cython to requirements to use pyximport
Browse files Browse the repository at this point in the history
  • Loading branch information
psbhatt committed Aug 26, 2024
1 parent 96ebe75 commit 64e4dbb
Show file tree
Hide file tree
Showing 4 changed files with 22 additions and 13 deletions.
3 changes: 2 additions & 1 deletion requirements.txt
Original file line number Diff line number Diff line change
@@ -1,3 +1,4 @@
numpy >= 1.24.1
scipy >= 1.4.1
docopt >= 0.6.2
docopt >= 0.6.2
Cython >= 3.0.0
12 changes: 9 additions & 3 deletions src/icikt/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -9,9 +9,15 @@
from os import path as _path
__version__ = _gv(_path.join(_path.dirname(__file__), _path.pardir))


from icikt.methods import icikt
from icikt.methods import iciktArray
from icikt.left_censorship import leftCensorTest
from icikt.utility import setupMissingMatrix

# from icikt.utility import setupMissingMatrix
#
#
# # try:
# # from icikt import c_kendall_dis
# # except ImportError:
# # # from icikt import kendall_dis as _kendall_dis
# # from icikt import kendall_dis as c_kendall_dis
# from icikt import c_kendall_dis
13 changes: 7 additions & 6 deletions src/icikt/__main__.py
Original file line number Diff line number Diff line change
Expand Up @@ -24,9 +24,10 @@
"""

from icikt.methods import iciktArray
from icikt.left_censorship import leftCensorTest
from icikt.__init__ import __version__ as ver
# from icikt.methods import iciktArray
# from icikt.left_censorship import leftCensorTest
# from icikt import __version__ as ver
import icikt
import multiprocessing
import docopt
import numpy as np
Expand All @@ -38,7 +39,7 @@ def main():
args = docopt.docopt(__doc__)

if args["--version"]:
print(ver)
print(icikt.__version__)

if args["leftCensor"]:

Expand Down Expand Up @@ -69,7 +70,7 @@ def main():
sys.exit(1)

try:
results = leftCensorTest(args["<dataFilePath>"], globalNA=args["--replace"], sampleClasses=args["--samples"])
results = icikt.leftCensorTest(args["<dataFilePath>"], globalNA=args["--replace"], sampleClasses=args["--samples"])
print(results)
except Exception as e:
print(e)
Expand Down Expand Up @@ -122,7 +123,7 @@ def main():
sys.exit(1)

try:
out, corr, pVal, tMax = iciktArray(dataArray=args["<dataFilePath>"], globalNA=args["--replace"], perspective=args["--mode"], scaleMax=args["--scale"], diagGood=args["--diag"], chunkSize=int(args["--chunk"]), includeOnly=args['--include'])
out, corr, pVal, tMax = icikt.iciktArray(dataArray=args["<dataFilePath>"], globalNA=args["--replace"], perspective=args["--mode"], scaleMax=args["--scale"], diagGood=args["--diag"], chunkSize=int(args["--chunk"]), includeOnly=args['--include'])

if args["--output"] is not None:
np.savetxt(args["--output"]+'outArray.csv', out, delimiter=',')
Expand Down
7 changes: 4 additions & 3 deletions src/icikt/methods.py
Original file line number Diff line number Diff line change
Expand Up @@ -18,11 +18,12 @@
from scipy.stats import distributions
from icikt.utility import setupMissingMatrix

import pyximport
pyximport.install()
try:
from icikt import c_kendall_dis
except ImportError:
# from icikt import kendall_dis as _kendall_dis
from icikt.kendall_dis import kendall_dis as c_kendall_dis
from icikt import kendall_dis as c_kendall_dis


def initialize_global_data(data):
Expand Down Expand Up @@ -129,7 +130,7 @@ def normtestFinish(z: float) -> tuple:
x, y = x[perm], y[perm]
x = np.r_[True, x[1:] != x[:-1]].cumsum(dtype=np.intp)

dis = _kendall_dis.kendall_dis(x, y) # discordant pairs
dis = c_kendall_dis.kendall_dis(x, y) # discordant pairs

obs = np.r_[True, (x[1:] != x[:-1]) | (y[1:] != y[:-1]), True]
cnt = np.diff(np.nonzero(obs)[0]).astype('int64', copy=False)
Expand Down

0 comments on commit 64e4dbb

Please sign in to comment.