Commit e44196f2 authored by Amir MOHAMMADI's avatar Amir MOHAMMADI

Fix tests

parent debad5f9
Pipeline #30785 passed with stage
in 20 minutes and 3 seconds
......@@ -10,6 +10,7 @@ class Predictions(Algorithm):
super(Predictions, self).__init__(**kwargs)
def score(self, predictions):
predictions = numpy.asarray(predictions)
if predictions.size == 1:
# output of a sigmoid binary layer
return predictions
......
......@@ -138,9 +138,7 @@ def test_video_gmm_pad_algorithm():
# Train the OneClassGMM machine and get normalizers:
machine, features_mean, features_std = algorithm.train_gmm(
real=real_array_converted,
n_components=algorithm.n_components,
random_state=algorithm.random_state)
real=real_array_converted)
algorithm.machine = machine
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment