From 522b68978028573f4b666d2ce9480ab9b6738de2 Mon Sep 17 00:00:00 2001
From: Haibao Tang <tanghaibao@gmail.com>
Date: Sun, 28 Apr 2024 12:02:37 -0700
Subject: [PATCH] Add params

---
 jcvi/graphics/landscape.py | 29 +++++++++++++++--------------
 1 file changed, 15 insertions(+), 14 deletions(-)

diff --git a/jcvi/graphics/landscape.py b/jcvi/graphics/landscape.py
index b28f889e..0c97e4be 100644
--- a/jcvi/graphics/landscape.py
+++ b/jcvi/graphics/landscape.py
@@ -11,6 +11,7 @@
 import sys
 
 from collections import Counter, OrderedDict, defaultdict
+from typing import Optional
 
 import numpy as np
 
@@ -22,17 +23,17 @@
 from ..utils.cbook import human_size, autoscale
 
 from .base import (
-    plt,
-    Rectangle,
     CirclePolygon,
-    savefig,
-    ticker,
+    Rectangle,
+    adjust_spines,
     human_readable_base,
     latex,
     markup,
-    set_human_axis,
     normalize_axes,
-    adjust_spines,
+    plt,
+    savefig,
+    set_human_axis,
+    ticker,
 )
 
 
@@ -269,14 +270,14 @@ def mosdepth(args):
 def draw_depth(
     root,
     ax,
-    bed,
-    chrinfo={},
-    defaultcolor="k",
-    sepcolor="w",
-    ylim=100,
-    logscale=False,
-    title=None,
-    subtitle=None,
+    bed: Bed,
+    chrinfo: dict = {},
+    defaultcolor: str = "k",
+    sepcolor: str = "w",
+    ylim: int = 100,
+    logscale: bool = False,
+    title: Optional[str] = None,
+    subtitle: Optional[str] = None,
 ):
     """Draw depth plot on the given axes, using data from bed