always return filtered stats, but will need more detailed statistics …

…and control logic eventually
lucidrains · Apr 7, 2023 · 4dbd33b · 4dbd33b
1 parent 8dd2a25
commit 4dbd33b
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/setup.py b/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',

diff --git a/toolformer_pytorch/toolformer_pytorch.py b/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