attempt fix live output post request handling

numerai · Oct 22, 2024 · f2f079f · f2f079f
1 parent 999cd78
commit f2f079f
Showing 1 changed file with 9 additions and 9 deletions.
diff --git a/predict.py b/predict.py
@@ -242,16 +242,16 @@ def main(args):
         logging.info("Uploading predictions to %s", args.post_url)
         csv_buffer = io.StringIO()
         predictions.to_csv(csv_buffer)
-        retry_request_with_backoff(
-            lambda: (
-                csv_buffer.seek(0)
-                and requests.post(
-                    args.post_url,
-                    data=args.post_data,
-                    files={"file": (predictions_csv_file_name, csv_buffer, "text/csv")},
-                )
+
+        def post_live_output():
+            csv_buffer.seek(0)
+            return requests.post(
+                args.post_url,
+                data=args.post_data,
+                files={"file": (predictions_csv_file_name, csv_buffer, "text/csv")},
             )
-        )
+
+        retry_request_with_backoff(post_live_output)
     else:
         logging.info("Saving predictions to %s", predictions_csv_file_name)
         with open(predictions_csv_file_name, "w") as f: