update

Author: rasbt

ch05/12_gemma3/tests/test_gemma3_kv_nb.py

@@ -18,7 +18,7 @@
 @pytest.fixture
 def nb_imports():
     nb_dir = Path(__file__).resolve().parents[1]
-    mod = import_definitions_from_notebook(nb_dir, "standalone-gemma3.ipynb")
+    mod = import_definitions_from_notebook(nb_dir, "standalone-gemma3-plus-kvcache.ipynb")
     return mod
 
 

ch05/12_gemma3/tests/test_gemma3_nb.py

@@ -18,7 +18,7 @@
 @pytest.fixture
 def nb_imports():
     nb_dir = Path(__file__).resolve().parents[1]
-    mod = import_definitions_from_notebook(nb_dir, "standalone-gemma3-plus-kv-cache.ipynb")
+    mod = import_definitions_from_notebook(nb_dir, "standalone-gemma3.ipynb")
     return mod