always return filtered stats, but will need more detailed statistics and control logic eventually

Author: lucidrains

setup.py

@@ -3,7 +3,7 @@
 setup(
   name = 'toolformer-pytorch',
   packages = find_packages(exclude=[]),
-  version = '0.0.24',
+  version = '0.0.25',
   license='MIT',
   description = 'Toolformer - Pytorch',
   author = 'Phil Wang',

toolformer_pytorch/toolformer_pytorch.py

@@ -876,7 +876,7 @@ def forward(
         data_with_responses = self.make_api_calls(filtered_data_with_api_calls)
         filtered_results = self.filter_by_api_responses(filtered_data, filtered_data_with_api_calls, data_with_responses)
 
-        if not self.should_finetune:
-            return filtered_results
+        if self.should_finetune:
+            self.finetune(filtered_results)
 
-        self.finetune(filtered_results)
+        return filtered_results