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",