Koza doesn't play nicely with generators

[DnlRKorn]

I tried to replace the following line of code

while (row := koza_app.get_row()) is not None:
xyz

with

koza_app = get_koza_app("go_annotation")
def row_gen():   while True:
       try:
           row = koza_app.get_row()
           yield row
       except StopIteration:
           return

for row in row_gen():
   xyz

But this caused some form of infinite loop to begin firing within the koza module

2024-08-16_14:09:34 | INFO | koza.io.reader.csv_reader | Finished processing 19 rows for go_annotation from file data/go/9606.go_annotations.gaf.gz
2024-08-16_14:09:34 | INFO | koza.io.reader.csv_reader | Finished processing 19 rows for go_annotation from file data/go/9606.go_annotations.gaf.gz
2024-08-16_14:09:34 | INFO | koza.io.reader.csv_reader | Finished processing 19 rows for go_annotation from file data/go/9606.go_annotations.gaf.gz
2024-08-16_14:09:34 | INFO | koza.io.reader.csv_reader | Finished processing 19 rows for go_annotation from file data/go/9606.go_annotations.gaf.gz
2024-08-16_14:09:34 | INFO | koza.io.reader.csv_reader | Finished processing 19 rows for go_annotation from file data/go/9606.go_annotations.gaf.gz

[ptgolden]

I'm fairly sure the issue here is that koza expects StopIteration to be raised to know when to stop running the transform script. Since that exception is never raised, the script continues to be run

[ptgolden]

i.e. it never breaks out of this loop:

koza/src/koza/app.py

Lines 102 to 119 in fceafe5

	while True:
	try:
	if is_first:
	transform_module = importlib.import_module(transform_code)
	is_first = False
	else:
	importlib.reload(transform_module)
	except MapItemException as mie:
	if self.logger:
	self.logger.debug(f"{str(mie)} not found in map")
	except NextRowException:
	continue
	except ValidationError:
	if self.logger:
	self.logger.error(f"Validation error while processing: {self.source.last_row}")
	raise ValidationError
	except StopIteration:
	break