ci: fix tests and lint prerequisites

Author: beinan

.github/workflows/python-test.yml

@@ -32,10 +32,16 @@ jobs:
           python-version: ${{ matrix.python-version }}
       - name: Setup Rust
         uses: actions-rust-lang/setup-rust-toolchain@v1
+        with:
+          cache: false
       - uses: Swatinem/rust-cache@v2
         with:
           workspaces: python
           cache-targets: false
+      - name: Install dependencies
+        run: |
+          sudo apt update
+          sudo apt install -y protobuf-compiler
       - name: Install uv
         uses: astral-sh/setup-uv@v3
       - name: Create virtual environment and install dependencies

.github/workflows/rust-test.yml

@@ -39,6 +39,10 @@ jobs:
         with:
           workspaces: rust/lance-context
           cache-targets: false
+      - name: Install dependencies
+        run: |
+          sudo apt update
+          sudo apt install -y protobuf-compiler
       - name: Build tests
         run: cargo test --manifest-path rust/lance-context/Cargo.toml --no-run
       - name: Run unit tests

python/python/lance_context/__init__.py

@@ -1,6 +1,7 @@
 from __future__ import annotations
 
-from ._internal import Context, version as _version
+from ._internal import Context
+from ._internal import version as _version
 
 __all__ = ["Context", "__version__"]