Skip to content

Commit

Permalink
set generator back to previous state, update var name (#43)
Browse files Browse the repository at this point in the history
Co-authored-by: John Myers <john@gretel.ai>
  • Loading branch information
johntmyers and John Myers committed Aug 4, 2020
1 parent 4d63b79 commit 56c065e
Showing 1 changed file with 4 additions and 5 deletions.
9 changes: 4 additions & 5 deletions src/gretel_synthetics/generator.py
Original file line number Diff line number Diff line change
Expand Up @@ -162,8 +162,8 @@ def generate_next(self, num_lines: int) -> Iterable[gen_text]:
Yields:
A ``gen_text`` object for every line (valid or invalid) that is generated.
"""
lines_generated = 0
while lines_generated < num_lines:
valid_lines_generated = 0
while valid_lines_generated < num_lines:
rec = _predict_chars(self.model, self.sp, self.settings.start_string, self.settings.config).data
_valid = None
try:
Expand All @@ -180,14 +180,13 @@ def generate_next(self, num_lines: int) -> Iterable[gen_text]:
except Exception as err:
# NOTE: this catches any exception raised by the line validator, which
# also creates an invalid record
lines_generated += 1
self.total_invalid += 1
yield gen_text(text=rec, valid=False, explain=str(err), delimiter=self.delim)
else:
if self.settings.line_validator and _valid:
lines_generated += 1
valid_lines_generated += 1
elif not self.settings.line_validator:
lines_generated += 1
valid_lines_generated += 1
else:
...

Expand Down

0 comments on commit 56c065e

Please sign in to comment.