From 61f2e19795cda2e922839351adce78d6047d3d86 Mon Sep 17 00:00:00 2001
From: tarantula-leo <54618933+tarantula-leo@users.noreply.github.com>
Date: Fri, 17 Nov 2023 11:18:22 +0800
Subject: [PATCH] Update classification_emul.py

---
 sml/metrics/classification/classification_emul.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/sml/metrics/classification/classification_emul.py b/sml/metrics/classification/classification_emul.py
index c023e366..3712abda 100644
--- a/sml/metrics/classification/classification_emul.py
+++ b/sml/metrics/classification/classification_emul.py
@@ -95,8 +95,8 @@ def check(spu_result, sk_result):
     # Test binary
     y_true = jnp.array([0, 1, 1, 0, 1, 1])
     y_pred = jnp.array([0, 0, 1, 0, 1, 1])
-    spu_result = emulator.run(proc)(y_true, y_pred, average=None, labels=[0, 1, 2])
-    sk_result = sklearn_proc(y_true, y_pred, average=None, labels=[0, 1, 2])
+    spu_result = emulator.run(proc)(y_true, y_pred, pos_label=1, transform=0)
+    sk_result = sklearn_proc(y_true, y_pred)
     check(spu_result, sk_result)
 
     # Test multiclass