Missed pieces

Lightning-AI · Mar 8, 2024 · c7fd796 · c7fd796
1 parent ba363b0
commit c7fd796
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/test_chat.py b/tests/test_chat.py
@@ -78,7 +78,7 @@ def decode(self, tensor: torch.Tensor) -> str:
     assert out.getvalue() == "baz bar foo "
 
 
-@patch("chat.base.input")
+@patch("litgpt.chat.base.input")
 @pytest.mark.parametrize("stop_iteration", [KeyboardInterrupt, ""])
 def test_main(mocked_input, stop_iteration, fake_checkpoint_dir, monkeypatch, tensor_like):
     import chat.base as chat
@@ -119,7 +119,7 @@ def test_main(mocked_input, stop_iteration, fake_checkpoint_dir, monkeypatch, te
 
 
 def test_cli():
-    cli_path = Path(__file__).parent.parent / "chat" / "base.py"
+    cli_path = Path(__file__).parent.parent / "litgpt/chat/base.py"
     output = subprocess.check_output([sys.executable, cli_path, "-h"])
     output = str(output.decode())
     assert "Starts a conversation" in output