MAINT: .sum() behaviour

This commit is contained in:
Gil Wassermann
2016-08-01 13:48:14 -04:00
parent c10af2a0b9
commit 7623c0f6eb
2 changed files with 2 additions and 3 deletions
+1 -1
View File
@@ -409,7 +409,7 @@ class FilterTestCase(BasePipelineTestCase):
window_length = 0
strictly_true_filter = StrictlyTrueFilter(
inputs=[InputFilter()],
inputs=(InputFilter(), ),
window_length=self.default_shape[0]
)
+1 -2
View File
@@ -499,6 +499,5 @@ class StrictlyTrueFilter(CustomFilter):
**Default Window Length**: None
"""
def compute(self, today, assets, out, arg):
out[:] = (sum(arg) == self.window_length)
out[:] = (arg.sum(axis=0) == self.window_length)