mirror of
https://github.com/wassname/catalyst.git
synced 2026-07-08 22:37:11 +08:00
MAINT: .sum() behaviour
This commit is contained in:
@@ -409,7 +409,7 @@ class FilterTestCase(BasePipelineTestCase):
|
||||
window_length = 0
|
||||
|
||||
strictly_true_filter = StrictlyTrueFilter(
|
||||
inputs=[InputFilter()],
|
||||
inputs=(InputFilter(), ),
|
||||
window_length=self.default_shape[0]
|
||||
)
|
||||
|
||||
|
||||
@@ -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)
|
||||
|
||||
Reference in New Issue
Block a user