From 84c5520ab8024f70087fd95b67f087952f320d2a Mon Sep 17 00:00:00 2001
From: mhchia <kevin.mh.chia@gmail.com>
Date: Fri, 13 Sep 2024 17:35:16 +0800
Subject: [PATCH] bump version to v0.1.10

---
 pyproject.toml     | 2 +-
 tests/conftest.py  | 6 ++++--
 tests/test_core.py | 4 +++-
 3 files changed, 8 insertions(+), 4 deletions(-)

diff --git a/pyproject.toml b/pyproject.toml
index 69b738d..0fb4515 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "zkstats"
-version = "0.1.9"
+version = "0.1.10"
 description = ""
 authors = ["Jern Kunpittaya", "Kevin Chia"]
 
diff --git a/tests/conftest.py b/tests/conftest.py
index 20788b0..08f9845 100644
--- a/tests/conftest.py
+++ b/tests/conftest.py
@@ -9,12 +9,14 @@ def error() -> float:
 
 @pytest.fixture
 def column_0():
-    return torch.tensor([3.0, 4.5, 1.0, 2.0, 7.5, 6.4, 5.5, 6.4])
+    # return torch.tensor([3.0, 4.5, 1.0, 2.0, 7.5, 6.4, 5.5, 6.4])
+    return torch.tensor([46.2, 40.4, 44.8, 48.1, 51.2, 91.9, 38.2])
 
 
 @pytest.fixture
 def column_1():
-    return torch.tensor([2.7, 3.3, 1.1, 2.2, 3.8, 8.2, 4.4, 3.8])
+    # return torch.tensor([2.7, 3.3, 1.1, 2.2, 3.8, 8.2, 4.4, 3.8])
+    return torch.tensor([3.5, 3.0, 3.3, 3.6, 3.8, 8.3, 3.1])
 
 
 @pytest.fixture
diff --git a/tests/test_core.py b/tests/test_core.py
index 71cc4ba..883d234 100644
--- a/tests/test_core.py
+++ b/tests/test_core.py
@@ -70,7 +70,9 @@ def test_integration_select_partial_columns(tmp_path, column_0, column_1, error,
     data_to_json_file(data_path, data_json)
 
     def simple_computation(state, args):
-        return state.mean(args["columns_0"])
+        m_0 = state.mean(args["columns_0"])
+        m_1 = state.mean(args["columns_1"])
+        return m_0, m_1
     precal_witness_path = tmp_path / "precal_witness_path.json"
     selected_columns, _, model = computation_to_model(simple_computation, precal_witness_path, data_shape, True, error)
     # gen settings, setup, prove, verify