Correct types

Author: w8jcik

src/moldrug/runner.py

@@ -6,7 +6,7 @@
 
 try:
     from dask.distributed import Client
-    from dask_jobqueue.core import JobQueueCluster
+    from dask_jobqueue import JobQueueCluster
 
     dask_available = True
 except ImportError:
@@ -47,8 +47,8 @@ class RunnerMode(Enum):
 
 class Runner:
     mode: RunnerMode
-    thread_count: int
-    process_count: int
+    thread_count: Optional[int]
+    process_count: Optional[int]
     dask_cluster = None
 
     def __init__(self, mode: RunnerMode, thread_count: Optional[int] = None, process_count: Optional[int] = None, dask_cluster = None):