more typing fixes, only tensor_utils and freeze left

mivanit · Feb 23, 2025 · 222dfe2 · 222dfe2
1 parent cd24e3b
commit 222dfe2
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 6 deletions.
diff --git a/tests/unit/nbutils/test_conversion.py b/tests/unit/nbutils/test_conversion.py
@@ -11,10 +11,10 @@
 from muutils.nbutils.convert_ipynb_to_script import process_dir, process_file
 from muutils.nbutils.run_notebook_tests import run_notebook_tests
 
-notebooks_input_dir: str = "tests/input_data/notebooks"
-notebooks_converted_input_dir: str = "tests/input_data/notebooks_converted/"
-nb_test_dir: str = "tests/_temp/run_notebook_tests"
-nb_conversion_dir: str = "tests/_temp/test_file_conversion"
+notebooks_input_dir: Path = Path("tests/input_data/notebooks")
+notebooks_converted_input_dir: Path = Path("tests/input_data/notebooks_converted/")
+nb_test_dir: Path = Path("tests/_temp/run_notebook_tests")
+nb_conversion_dir: Path = Path("tests/_temp/test_file_conversion")
 
 
 def test_run_notebook_tests():

diff --git a/tests/unit/test_parallel.py b/tests/unit/test_parallel.py
@@ -289,7 +289,7 @@ def test_parallel_performance(input_range, expected):
 @dataset_decorator(["small"])
 def test_reject_pbar_str_when_not_str_or_callable(input_range, expected):
     with pytest.raises(TypeError):
-        run_maybe_parallel(square, input_range, False, pbar=12345)
+        run_maybe_parallel(square, input_range, False, pbar=12345)  # type: ignore[arg-type]
 
 
 def custom_pbar(iterable: Iterable, **kwargs: Any) -> List:

diff --git a/tests/unit/test_spinner.py b/tests/unit/test_spinner.py
@@ -58,7 +58,7 @@ def long_running_function_with_status(
 
     # Run the example functions
     print("Running function with status updates:")
-    result1: str = long_running_function_with_status(1)
+    result1: str = long_running_function_with_status(1)  # type: ignore[call-arg]
     print(result1)