0.0.57: accept lists and series in constructor

pyranges · Oct 14, 2020 · 44f77a3 · 44f77a3
1 parent 5970b21
commit 44f77a3
Show file tree

Hide file tree

Showing 4 changed files with 1,864 additions and 1,445 deletions.
diff --git a/CHANGELOG b/CHANGELOG
@@ -1,3 +1,6 @@
+# 0.0.57 (14.10.20)
+- accept lists/pd.Series in constructor
+
 # 0.0.55-56 (12.10.20)
 - fix build on 32-bit architectures (thanks to nileshpatra)
 

diff --git a/ncls/__init__.py b/ncls/__init__.py
@@ -5,6 +5,9 @@
 
 def NCLS(starts, ends, ids):
 
+ if isinstance(starts, list) or "pandas" in str(type(starts)):
+ starts, ends, ids = [np.array(s) for s in [starts, ends, ids]]
+
  if starts.dtype == np.int64:
  return NCLS64(starts, ends, ids)
  elif starts.dtype == np.int32:
@@ -17,6 +20,9 @@ def FNCLS(starts, ends, ids):
 
  from ncls.src.fncls import FNCLS
 
+ if isinstance(starts, list) or "pandas" in str(type(starts)):
+ starts, ends, ids = [np.array(s) for s in [starts, ends, ids]]
+
  if starts.dtype == np.double:
  return FNCLS(starts, ends, ids)
  else: