improve code in check_intended_purpose

compliance_analysis.py

@@ -116,106 +116,55 @@ def run_compliance_analysis_on_model(model_cc_yaml):
    
     return msg
 
-def check_intended_use_aligned():
+def check_intended_purpose():
     
     # We want to run this function for everything classified as a high_risk_ai_system
     # We also need to run it for all 
     # Add any of the intended purposes of the overall project to a set of intended purposes
     
-    if project_cc_yaml['high_risk_ai_system']['safety_component'] == True:
-        intended_purpose = "safety_component"
-    if project_cc_yaml['high_risk_ai_system']['product_regulated_machinery'] == True:
-        intended_purpose = "product_regulated_machinery" 
-    if project_cc_yaml['high_risk_ai_system']['product_regulated_toy'] == True:
-        intended_purpose = "product_regulated_toy" 
-    if project_cc_yaml['high_risk_ai_system']['product_regulated_watercraft'] == True:
-        intended_purpose = "product_regulated_watercraft" 
-    if project_cc_yaml['high_risk_ai_system']['biometric_categorization'] == True:
-        intended_purpose = "biometric_categorization" 
-    if project_cc_yaml['high_risk_ai_system']['emotion_recognition'] == True:
-        intended_purpose = "emotion_recognition" 
-    if project_cc_yaml['high_risk_ai_system']['critical_infrastructure'] == True:
-        intended_purpose = "critical_infrastructure" 
-    if project_cc_yaml['high_risk_ai_system']['admission'] == True:
-        intended_purpose = "admission" 
-    if project_cc_yaml['high_risk_ai_system']['recruitment'] == True:
-        intended_purpose = "recruitment" 
-    if project_cc_yaml['high_risk_ai_system']['public_assistance'] == True:
-        intended_purpose = "public_assistance"
-    if project_cc_yaml['high_risk_ai_system']['victim_assessment'] == True:
-        intended_purpose = "victim_assessment" 
-    if project_cc_yaml['high_risk_ai_system']['polygraph'] == True:
-        intended_purpose = "polygraph" 
-    if project_cc_yaml['high_risk_ai_system']['judicial'] == True:
-        intended_purpose = "judicial" 
+    # intended_purpose = ['safety_component',
+    #                     "product_regulated_machinery",
+    #                     "product_regulated_toy",
+    #                     "product_regulated_watercraft",
+    #                     "biometric_categorization",
+    #                     "emotion_recognition",
+    #                     "critical_infrastructure",
+    #                     "admission",
+    #                     "recruitment",
+    #                     "public_assistance",
+    #                     "victim_assessment",
+    #                     "polygraph",
+    #                     "judicial"]
+    
+    project_intended_purposes = []
+    for key in project_cc_yaml['high_risk_ai_system']:
+        if project_cc_yaml['high_risk_ai_system'][f'{key}']['value']:
+            project_intended_purposes.append(key) 
     
     # For each Data CC, put the intended uses in a set and then make sure the Project's intended use is in the set
 
-    dataset_intended_purpose = () 
-
-    if data_cc_yaml['intended_purpose']['safety_component'] == True:
-        dataset_intended_purpose.add("safety_component")
-    if data_cc_yaml['intended_purpose']['product_regulated_machinery'] == True:
-        dataset_intended_purpose.add("product_regulated_machinery")
-    if data_cc_yaml['intended_purpose']['product_regulated_toy'] == True:
-        dataset_intended_purpose.add("product_regulated_toy")
-    if data_cc_yaml['intended_purpose']['product_regulated_watercraft'] == True:
-        dataset_intended_purpose.add("product_regulated_watercraft")
-    if data_cc_yaml['intended_purpose']['biometric_categorization'] == True:
-        dataset_intended_purpose.add("biometric_categorization")
-    if data_cc_yaml['intended_purpose']['emotion_recognition'] == True:
-        dataset_intended_purpose.add("emotion_recognition")
-    if data_cc_yaml['intended_purpose']['critical_infrastructure'] == True:
-        dataset_intended_purpose.add("critical_infrastructure")
-    if data_cc_yaml['intended_purpose']['admission'] == True:
-        dataset_intended_purpose.add("admission")
-    if data_cc_yaml['intended_purpose']['recruitment'] == True:
-        dataset_intended_purpose.add("recruitment")
-    if data_cc_yaml['intended_purpose']['public_assistance'] == True:
-        dataset_intended_purpose.add("public_assistance")       
-    if data_cc_yaml['intended_purpose']['victim_assessment'] == True:
-        dataset_intended_purpose.add("victim_assessment")
-    if data_cc_yaml['intended_purpose']['polygraph'] == True:
-        dataset_intended_purpose.add("polygraph")
-    if data_cc_yaml['intended_purpose']['judicial'] == True:
-        dataset_intended_purpose.add("judicial")
-
-    if project_intended_purpose not in dataset_intended_purpose:
-        print("You are not compliant")
+    msg = ''
+    dataset_intended_purposes = []
+    for key in data_cc_yaml['high_risk_ai_system']:
+        if data_cc_yaml['high_risk_ai_system'][f'{key}']['value']:
+            dataset_intended_purposes.append(key) 
+
+    for purpose in project_intended_purposes:
+        if purpose not in dataset_intended_purposes:
+            msg = f"You are not compliant because {purpose} is not a valid purpose"
 
     # Now do the exact same thing for all models
 
-    model_intended_purpose = () 
-
-    if model_cc_yaml['intended_purpose']['safety_component'] == True:
-        dataset_intended_purpose.add("safety_component")
-    if model_cc_yaml['intended_purpose']['product_regulated_machinery'] == True:
-        dataset_intended_purpose.add("product_regulated_machinery")
-    if model_cc_yaml['intended_purpose']['product_regulated_toy'] == True:
-        dataset_intended_purpose.add("product_regulated_toy")
-    if model_cc_yaml['intended_purpose']['product_regulated_watercraft'] == True:
-        dataset_intended_purpose.add("product_regulated_watercraft")
-    if model_cc_yaml['intended_purpose']['biometric_categorization'] == True:
-        dataset_intended_purpose.add("biometric_categorization")
-    if model_cc_yaml['intended_purpose']['emotion_recognition'] == True:
-        dataset_intended_purpose.add("emotion_recognition")
-    if model_cc_yaml['intended_purpose']['critical_infrastructure'] == True:
-        dataset_intended_purpose.add("critical_infrastructure")
-    if model_cc_yaml['intended_purpose']['admission'] == True:
-        dataset_intended_purpose.add("admission")
-    if model_cc_yaml['intended_purpose']['recruitment'] == True:
-        dataset_intended_purpose.add("recruitment")
-    if model_cc_yaml['intended_purpose']['public_assistance'] == True:
-        dataset_intended_purpose.add("public_assistance")       
-    if model_cc_yaml['intended_purpose']['victim_assessment'] == True:
-        dataset_intended_purpose.add("victim_assessment")
-    if model_cc_yaml['intended_purpose']['polygraph'] == True:
-        dataset_intended_purpose.add("polygraph")
-    if model_cc_yaml['intended_purpose']['judicial'] == True:
-        dataset_intended_purpose.add("judicial")
-
-    if project_intended_purpose not in dataset_intended_purpose:
-        print("You are not compliant")
+    model_intended_purposes = []
+    for key in model_cc_yaml['high_risk_ai_system']:
+        if model_cc_yaml['high_risk_ai_system'][f'{key}']['value']:
+            model_intended_purposes.append(key) 
+
+    for purpose in project_intended_purposes:
+        if purpose not in model_intended_purposes:
+            msg = f"You are not compliant because {purpose} is not a valid purpose"
+
+    return msg
 
 
 

data_cc.yaml

@@ -19,7 +19,7 @@ intended_purpose:
     value: !!bool false 
   critical_infrastructure: # Art. 6(2); Annex III(2)
     verbose: 'This dataset is appropriate to use for AI projects involving safety components in the management and operation of critical digital infrastructure, road traffic, or in the supply of water, gas, heating or electricity'
-    value: !!bool false 
+    value: !!bool true 
   admission: # Art. 6(2); Annex III(3)(a)
     verbose: 'This dataset is appropriate to use for AI projects involving the determination of access or admission or to assigning natural persons to educational and vocational training institutions at all levels'
     value: !!bool false 

project_cc.yaml

@@ -152,10 +152,10 @@ high_risk_ai_system:
     value: !!bool false 
   emotion_recognition: # Art. 6(2); Annex III(1)(c)
     verbose: 'AI project is intended to be used for emotion recognition'
-    value: !!bool false 
+    value: !!bool true 
   critical_infrastructure: # Art. 6(2); Annex III(2)
     verbose: 'AI project is intended to be used as a safety component in the management and operation of critical digital infrastructure, road traffic, or in the supply of water, gas, heating or electricity'
-    value: !!bool false 
+    value: !!bool true 
   admission: # Art. 6(2); Annex III(3)(a)
     verbose: 'AI project is intended to be used to determine access or admission or to assign natural persons to educational and vocational training institutions at all levels'
     value: !!bool false