upate

Signed-off-by: Chi Zhang <czhangseu@gmail.com>

Author: vermouth1992

.github/workflows/tutorial-check.yml

@@ -33,8 +33,14 @@ jobs:
         python -m pip install --upgrade pip
         pip install torch torchvision --index-url https://download.pytorch.org/whl/cpu
         pip install -e ".[yuanrong]"
-    - name: Run tutorials
+        pip install nbconvert ipykernel
+    - name: Run Python tutorials
       run: |
         export TQ_NUM_THREADS=2
         export RAY_DEDUP_LOGS=0
-        for file in tutorial/*.py; do python3 "$file"; done
+        for file in tutorial/*.py; do python3 "$file"; done
+    - name: Run notebook tutorials
+      run: |
+        export TQ_NUM_THREADS=2
+        export RAY_DEDUP_LOGS=0
+        jupyter nbconvert --to notebook --execute --ExecutePreprocessor.timeout=120 tutorial/basic.ipynb

tests/e2e/test_kv_interface_e2e.py

@@ -182,9 +182,7 @@ def assert_nested_tensor_equal(nested_a, nested_b, msg=""):
     """Assert two nested (jagged) tensors are equal component-wise."""
     components_a = list(nested_a)
     components_b = list(nested_b)
-    assert len(components_a) == len(components_b), (
-        f"{msg} Length mismatch: {len(components_a)} vs {len(components_b)}"
-    )
+    assert len(components_a) == len(components_b), f"{msg} Length mismatch: {len(components_a)} vs {len(components_b)}"
     for i, (a, b) in enumerate(zip(components_a, components_b, strict=True)):
         assert torch.equal(a, b), f"{msg} Component {i} not equal: {a} vs {b}"
 
@@ -529,7 +527,6 @@ def test_kv_batch_put_returns_cumulative_fields(self, controller, tq_api):
 class TestKVGetE2E:
     """End-to-end tests for kv_batch_get functionality."""
 
-
     def test_kv_batch_get_nested_tensor(self, controller, tq_api):
         partition_id = "test_partition"
         keys = []
@@ -548,7 +545,6 @@ def test_kv_batch_get_nested_tensor(self, controller, tq_api):
         assert_nested_tensor_equal(retrieved["data"], torch.nested.as_nested_tensor(data_list, layout=torch.jagged))
         tq_api.kv_clear(keys=keys, partition_id=partition_id)
 
-
     def test_kv_batch_get_single_key(self, controller, tq_api):
         """Test getting data for a single key."""
         partition_id = "test_partition"