about summary refs log tree commit diff
diff options
context:
space:
mode:
-rw-r--r--gn3/db/genotypes.py10
1 files changed, 5 insertions, 5 deletions
diff --git a/gn3/db/genotypes.py b/gn3/db/genotypes.py
index 6f867c7..81efda2 100644
--- a/gn3/db/genotypes.py
+++ b/gn3/db/genotypes.py
@@ -2,12 +2,12 @@
 
 import os
 import gzip
+from typing import Union
+from pathlib import Path
 
-from gn3.settings import GENOTYPE_FILES
-
-def build_genotype_file(
-        geno_name: str, base_dir: str = GENOTYPE_FILES,
-        extension: str = "geno"):
+def build_genotype_file(geno_name: str,
+                        base_dir: Union[str, Path],
+                        extension: str = "geno"):
     """Build the absolute path for the genotype file."""
     return f"{os.path.abspath(base_dir)}/{geno_name}.{extension}"