change signature for in to match

cosmin · Feb 1, 2021 · d08fb05 · d08fb05
1 parent f03ebd7
commit d08fb05
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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)