mypy fixes

Author: stevebachmeier

setup.py

@@ -56,6 +56,7 @@
         "types-tqdm",
         "types-setuptools",
         "pyarrow-stubs",
+        "types-psutil",
     ]
 
     setup_requires = ["setuptools_scm"]

src/pseudopeople/interface.py

@@ -942,15 +942,15 @@ def set_up_dask_client() -> None:
     # Determine whether or not a Dask client is already running. If not,
     # create a new one.
     try:
-        client = get_client()
+        get_client()
     except ValueError:
         # No Dask client is running so we create one.
         from dask.distributed import LocalCluster
         from dask.system import CPU_COUNT
 
         # extract the memory limit from the environment variable
-        cluster = LocalCluster(
+        cluster = LocalCluster(  # type: ignore [no-untyped-call]
             n_workers=CPU_COUNT,
             threads_per_worker=1,
         )
-        client = cluster.get_client()
+        cluster.get_client()  # type: ignore [no-untyped-call]

tests/unit/test_interface.py

@@ -111,7 +111,7 @@ def test_set_up_dask_client_default() -> None:
 
     set_up_dask_client()
     client = get_client()
-    workers = client.scheduler_info()["workers"]
+    workers = client.scheduler_info()["workers"]  # type: ignore[no-untyped-call]
     assert len(workers) == CPU_COUNT
     assert all(worker["nthreads"] == 1 for worker in workers.values())
     if is_on_slurm():
@@ -134,13 +134,13 @@ def test_set_up_dask_client_default() -> None:
 def test_set_up_dask_client_custom() -> None:
     memory_limit = 1  # gb
     n_workers = 3
-    cluster = LocalCluster(
+    cluster = LocalCluster(  # type: ignore[no-untyped-call]
         name="custom",
         n_workers=n_workers,
         threads_per_worker=2,
         memory_limit=memory_limit * 1024**3,
     )
-    client = cluster.get_client()
+    client = cluster.get_client()  # type: ignore[no-untyped-call]
     set_up_dask_client()
     client = get_client()
     assert client.cluster.name == "custom"