Skip to content

Commit 2e34ff1

Browse files
jonbrenasjonbrenas
committed
Going the opposite way.
1 parent f141953 commit 2e34ff1

1 file changed

Lines changed: 71 additions & 71 deletions

File tree

‎malariagen_data/anoph/h12.py‎

Lines changed: 71 additions & 71 deletions
Original file line numberDiff line numberDiff line change
@@ -479,78 +479,78 @@ def plot_h12_gwss_track(
479479
for i in range(len(circle_kwargs_param), 5)
480480
}
481481
)
482-
elif isinstance(circle_kwargs_param, dict[str, dict]):
483-
if list(circle_kwargs_param.keys())[0] in [
484-
"2L",
485-
"2R",
486-
"3L",
487-
"3R",
488-
"X",
489-
"2RL",
490-
"3RL",
491-
]:
492-
for i in range(0, 5):
493-
if i == 0:
494-
if "2L" in circle_kwargs_param.keys(): # Ag3
495-
circle_kwargs_param_dict[
496-
i
497-
] = gplt_params.default_circle_kwargs_dict[
498-
i
499-
] # circle_kwargs_param["2L"]
500-
elif "2RL" in circle_kwargs_param.keys(): # Af1
501-
circle_kwargs_param_dict[
502-
i
503-
] = gplt_params.default_circle_kwargs_dict[
504-
i
505-
] # circle_kwargs_param["2RL"]
506-
else:
507-
circle_kwargs_param_dict[
508-
i
509-
] = gplt_params.default_circle_kwargs_dict[i]
510-
elif i == 1:
511-
if "2R" in circle_kwargs_param.keys(): # Ag3
512-
circle_kwargs_param_dict[i] = circle_kwargs_param["2R"]
513-
elif "3RL" in circle_kwargs_param.keys(): # Af1
514-
circle_kwargs_param_dict[i] = circle_kwargs_param["3RL"]
515-
else:
516-
circle_kwargs_param_dict[
517-
i
518-
] = gplt_params.default_circle_kwargs_dict[i]
519-
elif i == 2:
520-
if "3L" in circle_kwargs_param.keys(): # Ag3
521-
circle_kwargs_param_dict[i] = circle_kwargs_param["3L"]
522-
elif "X" in circle_kwargs_param.keys(): # Af1
523-
circle_kwargs_param_dict[i] = circle_kwargs_param["X"]
524-
else:
525-
circle_kwargs_param_dict[
526-
i
527-
] = gplt_params.default_circle_kwargs_dict[i]
528-
elif i == 3:
529-
if "3R" in circle_kwargs_param.keys(): # Ag3
530-
circle_kwargs_param_dict[i] = circle_kwargs_param["3R"]
531-
else:
532-
circle_kwargs_param_dict[
533-
i
534-
] = gplt_params.default_circle_kwargs_dict[i]
535-
elif i == 4:
536-
if (
537-
"X" in circle_kwargs_param.keys()
538-
): # Ag3. Will also get a value for Af1 but it will be ignored.
539-
circle_kwargs_param_dict[i] = circle_kwargs_param["X"]
540-
else:
541-
circle_kwargs_param_dict[
542-
i
543-
] = gplt_params.default_circle_kwargs_dict[i]
544-
else:
545-
circle_kwargs_param_dict = {i: circle_kwargs_param for i in range(0, 5)}
546-
elif isinstance(circle_kwargs_param, dict[int, dict]):
547-
for i in range(0, 5):
548-
if i in list(circle_kwargs_param.keys()):
549-
circle_kwargs_param_dict[i] = circle_kwargs_param[i]
482+
elif isinstance(circle_kwargs_param, dict):
483+
if isinstance(list(circle_kwargs_param.keys())[0], str):
484+
if list(circle_kwargs_param.keys())[0] in [
485+
"2L",
486+
"2R",
487+
"3L",
488+
"3R",
489+
"X",
490+
"2RL",
491+
"3RL",
492+
]:
493+
for i in range(0, 5):
494+
if i == 0:
495+
ck = circle_kwargs_param["2L"]
496+
if "2L" in circle_kwargs_param.keys(): # Ag3
497+
circle_kwargs_param_dict[i] = ck
498+
elif "2RL" in circle_kwargs_param.keys(): # Af1
499+
circle_kwargs_param_dict[
500+
i
501+
] = gplt_params.default_circle_kwargs_dict[
502+
i
503+
] # circle_kwargs_param["2RL"]
504+
else:
505+
circle_kwargs_param_dict[
506+
i
507+
] = gplt_params.default_circle_kwargs_dict[i]
508+
elif i == 1:
509+
if "2R" in circle_kwargs_param.keys(): # Ag3
510+
circle_kwargs_param_dict[i] = circle_kwargs_param["2R"]
511+
elif "3RL" in circle_kwargs_param.keys(): # Af1
512+
circle_kwargs_param_dict[i] = circle_kwargs_param["3RL"]
513+
else:
514+
circle_kwargs_param_dict[
515+
i
516+
] = gplt_params.default_circle_kwargs_dict[i]
517+
elif i == 2:
518+
if "3L" in circle_kwargs_param.keys(): # Ag3
519+
circle_kwargs_param_dict[i] = circle_kwargs_param["3L"]
520+
elif "X" in circle_kwargs_param.keys(): # Af1
521+
circle_kwargs_param_dict[i] = circle_kwargs_param["X"]
522+
else:
523+
circle_kwargs_param_dict[
524+
i
525+
] = gplt_params.default_circle_kwargs_dict[i]
526+
elif i == 3:
527+
if "3R" in circle_kwargs_param.keys(): # Ag3
528+
circle_kwargs_param_dict[i] = circle_kwargs_param["3R"]
529+
else:
530+
circle_kwargs_param_dict[
531+
i
532+
] = gplt_params.default_circle_kwargs_dict[i]
533+
elif i == 4:
534+
if (
535+
"X" in circle_kwargs_param.keys()
536+
): # Ag3. Will also get a value for Af1 but it will be ignored.
537+
circle_kwargs_param_dict[i] = circle_kwargs_param["X"]
538+
else:
539+
circle_kwargs_param_dict[
540+
i
541+
] = gplt_params.default_circle_kwargs_dict[i]
550542
else:
551-
circle_kwargs_param_dict[
552-
i
553-
] = gplt_params.default_circle_kwargs_dict[i]
543+
circle_kwargs_param_dict = {
544+
i: circle_kwargs_param for i in range(0, 5)
545+
}
546+
elif isinstance(list(circle_kwargs_param.keys())[0], int):
547+
for i in range(0, 5):
548+
if i in list(circle_kwargs_param.keys()):
549+
circle_kwargs_param_dict[i] = circle_kwargs_param[i]
550+
else:
551+
circle_kwargs_param_dict[
552+
i
553+
] = gplt_params.default_circle_kwargs_dict[i]
554554
else:
555555
circle_kwargs_param_dict = gplt_params.default_circle_kwargs_dict
556556

0 commit comments

Comments
 (0)