fix for repeated restarts on HF

validation.py

@@ -32,12 +32,12 @@ def validate_croissant(json_data):
         error_message = f"Unexpected error during validation: {str(e)}\n\n{error_details}"
         return False, error_message
     
-def try_generate_record(record_iterator):
+def try_generate_record(record_collection):
     try:
-        next(record_iterator)
+        iterator = iter(record_collection)
+        next(iterator)
         return "success"
     except Exception as e:
-        # Return the exception object to the outer function
         return e
 
 def validate_records(json_data):
@@ -53,9 +53,10 @@ def validate_records(json_data):
         
         for record_set in record_sets:
             try:
-                records = dataset.records(record_set=record_set.uuid)
-
-                result = func_timeout.func_timeout(WAIT_TIME, try_generate_record, args=(iter(records),))
+                result = func_timeout.func_timeout(
+                    WAIT_TIME,
+                    lambda: try_generate_record(dataset.records(record_set=record_set.uuid))
+                )
 
                 if isinstance(result, Exception):
                     raise result  # re-raise actual error outside timeout