Merge pull request #58 from rcpch/truncate-who-child

fix-truncate-who-child

Author: eatyourpeas

.bumpversion.cfg

@@ -1,5 +1,5 @@
 [bumpversion]
-current_version = 4.3.1
+current_version = 4.3.2
 tag = False
 commit = True
 

rcpchgrowth/chart_functions.py

@@ -26,6 +26,7 @@
     TURNERS, 
     UK_WHO, 
     UK_WHO_REFERENCES, 
+    UK_WHO_CHILD,
     WHO,
     WHO_REFERENCES
 )
@@ -352,8 +353,8 @@ def create_uk_who_chart(
         except:
             lms_array_for_measurement = []
 
-        # truncate the who_child data to stop at 4y
-        if len(lms_array_for_measurement) > 0:
+        # truncate the who_child data to stop at 4y in the WHO child reference. Must not do this for the UK90 reference
+        if len(lms_array_for_measurement) > 0 and reference == UK_WHO_CHILD:
             lms_array_for_measurement = [obj for obj in lms_array_for_measurement if obj["decimal_age"] <= 4.0]
 
         for centile_index, centile_sds in enumerate(centile_sds_collection):