From 2725cbdf765c2f2c94d0fa28a4eaf10f73339f52 Mon Sep 17 00:00:00 2001 From: zachary62 Date: Sat, 25 Jan 2025 05:39:44 +0000 Subject: [PATCH] track the current retry --- pocketflow/__init__.py | 4 ++-- setup.py | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/pocketflow/__init__.py b/pocketflow/__init__.py index 32ed3dc..17271fa 100644 --- a/pocketflow/__init__.py +++ b/pocketflow/__init__.py @@ -27,10 +27,10 @@ class Node(BaseNode): def __init__(self,max_retries=1,wait=0): super().__init__();self.max_retries,self.wait=max_retries,wait def exec_fallback(self,prep_res,exc): raise exc def _exec(self,prep_res): - for i in range(self.max_retries): + for self.cur_retry in range(self.max_retries): try: return self.exec(prep_res) except Exception as e: - if i==self.max_retries-1: return self.exec_fallback(prep_res,e) + if self.cur_retry==self.max_retries-1: return self.exec_fallback(prep_res,e) if self.wait>0: time.sleep(self.wait) class BatchNode(Node): diff --git a/setup.py b/setup.py index bf339ce..6526f78 100644 --- a/setup.py +++ b/setup.py @@ -2,7 +2,7 @@ from setuptools import setup, find_packages setup( name="pocketflow", - version="0.0.0", + version="0.0.1", packages=find_packages(), author="Zachary Huang", author_email="zh2408@columbia.edu",