fix: tests

Signed-off-by: Avik Basu <[email protected]>
numaproj · Jul 25, 2023 · 777d451 · 777d451
1 parent 2196fc6
commit 777d451
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/models/test_threshold.py b/tests/models/test_threshold.py
@@ -68,7 +68,7 @@ def test_score_samples(self):
         clf = MahalanobisThreshold()
         clf.fit(self.x_train)
         y = clf.score_samples(self.x_test)
-        self.assertTupleEqual(self.x_test.shape, y.shape)
+        self.assertTupleEqual((self.x_test.shape[0],), y.shape)
 
     def test_score_samples_err(self):
         clf = MahalanobisThreshold()