Update custom_llm.py

custom_llm.py

@@ -166,7 +166,7 @@ def out_format(text, llm, df):
 
     e_ = None
 
-    for _ in range(6):
+    for i in range(6):
 
         try :
             # text = text.split("```python")[-1].split("```")[0]
@@ -176,7 +176,7 @@ def out_format(text, llm, df):
             
             return text
         except Exception as e:
-            print(e)
+            print(f"ERORRR! ATTEMPT : {i}\n",str(e))
             text = err_chain.invoke({"code":text, "err":str(e)})
             e_ = e
     # exec(text)