MAINT: Add type assertions for labelarray_where.

quantopian · Apr 27, 2020 · 3eb9a71 · 3eb9a71
1 parent 6ca266c
commit 3eb9a71
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/zipline/lib/labelarray.py b/zipline/lib/labelarray.py
@@ -825,6 +825,7 @@ def __lt__(self, other):
         return True
 
 
+@expect_types(trues=LabelArray, falses=LabelArray)
 def labelarray_where(cond, trues, falses):
     """LabelArray-aware implementation of np.where.
     """