From d08fb05a8b11ec8234fa9d6e6768e7b7e13450cd Mon Sep 17 00:00:00 2001 From: Cosmin Stejerean Date: Mon, 1 Feb 2021 11:40:58 -0800 Subject: [PATCH] change signature for in to match --- python/vmaf/core/niqe_train_test_model.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/python/vmaf/core/niqe_train_test_model.py b/python/vmaf/core/niqe_train_test_model.py index fa409039f..772949e96 100644 --- a/python/vmaf/core/niqe_train_test_model.py +++ b/python/vmaf/core/niqe_train_test_model.py @@ -28,13 +28,13 @@ def _assert_dimension(cls, feature_names, results): @classmethod @override(TrainTestModel) - def get_xs_from_results(cls, results, indexs=None, aggregate=False): + def get_xs_from_results(cls, results, indexs=None, aggregate=False, features=None): """ override by altering aggregate default to False """ return super(NiqeTrainTestModel, cls).get_xs_from_results( - results, indexs, aggregate) + results, indexs, aggregate, features) @classmethod @override(TrainTestModel)