From f788fc5e7ea1e4b352a0e0de2d923ab568f46024 Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Fri, 31 May 2024 20:15:04 +0000 Subject: [PATCH] [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --- .../jupyter_ai/tests/doc_loaders/test_eligible_files.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/packages/jupyter-ai/jupyter_ai/tests/doc_loaders/test_eligible_files.py b/packages/jupyter-ai/jupyter_ai/tests/doc_loaders/test_eligible_files.py index b07278982..445aa579e 100644 --- a/packages/jupyter-ai/jupyter_ai/tests/doc_loaders/test_eligible_files.py +++ b/packages/jupyter-ai/jupyter_ai/tests/doc_loaders/test_eligible_files.py @@ -5,6 +5,7 @@ """ import os + from jupyter_ai.document_loaders.directory import collect_files # Prepare temp directory for test with pytest.py @@ -31,6 +32,7 @@ filepath = os.path.join(d, file) open(filepath, "a") + # Test that the number of valid files for `/learn` is correct def test_collect_files(): all_files = False @@ -40,5 +42,6 @@ def test_collect_files(): # Clean up temp directory from shutil import rmtree + rmtree(path) - os.chdir(os.path.split(path)[0]) \ No newline at end of file + os.chdir(os.path.split(path)[0])