Merge pull request #443 from IDAS-Durham/fix_yaml_bug

adding yaml import to immunity_setter
IDAS-Durham · Jan 4, 2022 · ef22ec3 · ef22ec3
2 parents aebd062 + 9766306
commit ef22ec3
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/june/epidemiology/infection/immunity_setter.py b/june/epidemiology/infection/immunity_setter.py
@@ -1,5 +1,6 @@
 from typing import Optional
 import numpy as np
+import yaml
 from random import random
 from june.utils import (
     parse_age_probabilities,