Update tools.py

tools.py

@@ -55,9 +55,15 @@ class AudioEvaluationRequest(BaseEvaluationRequest):
 class evaluate_consumption_example(Tool):
     name = "evaluate_consumption_example"
     description = "This is only an example. If a manager wants to know what you are capable of, use it : it will use code carbon to evaluate the CO2 emissions from an example Python code"
+    inputs = {
+        "code": {
+            "type": "string",
+            "description": "The code to evaluate. Here, it is an example, so just set it to 'None'.
+        }
+    }
     output_type = "string"
     
-    def forward(self, code=None):
+    def forward(self, code :  string):
         request = AudioEvaluationRequest() 
         logging.info("Chargement des données")
         dataset = load_dataset(request.dataset_name, streaming=True, token=os.getenv("HF_TOKEN"))
@@ -126,7 +132,7 @@ class evaluate_consumption(Tool):
     }
     output_type = "string"
     
-    def forward(self, code):
+    def forward(self, code : string):
         
         # Start tracking emissions
         tracker.start()