From c1e488035f2d896a2c4a0e20639c44b381aa2081 Mon Sep 17 00:00:00 2001
From: Frederick Muriuki Muriithi
Date: Thu, 26 Oct 2023 07:17:38 +0300
Subject: Fix errors caught by mypy.

---
 gn3/computations/gemma.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

(limited to 'gn3/computations')

diff --git a/gn3/computations/gemma.py b/gn3/computations/gemma.py
index 8036a7b..7f09b0a 100644
--- a/gn3/computations/gemma.py
+++ b/gn3/computations/gemma.py
@@ -5,6 +5,7 @@ from base64 import b64encode
 from hashlib import md5
 from typing import Dict
 from typing import List
+from typing import Optional
 from typing import ValuesView
 from gn3.commands import compose_gemma_cmd
 from gn3.fs_helpers import get_hash_of_files
@@ -53,8 +54,8 @@ def generate_gemma_cmd(gemma_cmd: str,
                        output_dir: str,
                        token: str,
                        gemma_kwargs: Dict,
-                       gemma_wrapper_kwargs: Dict = None,
-                       chromosomes: str = None) -> Dict:
+                       gemma_wrapper_kwargs: Optional[Dict] = None,
+                       chromosomes: Optional[str] = None) -> Dict:
     """Compute k values"""
     _hash = get_hash_of_files(
         [v for k, v in gemma_kwargs.items() if k in ["g", "p", "a", "c"]])
-- 
cgit v1.2.3