bundle models have arrived

This commit is contained in:
Josh Burman
2021-12-17 00:55:27 +00:00
parent dc696b47ac
commit 95aad63db6
5 changed files with 43 additions and 23 deletions

View File

@ -15,6 +15,7 @@ class LoftService(Base):
def process(self):
try:
self.solver_run = SolverRun.parse_obj(self.retreive_attributes_from_message())
self.solver_run.generate_bundles()
self.solution = self.generate_solution()
self.result = self.stream_to_s3_bucket()
except ItemGenerationError as error:
@ -40,7 +41,7 @@ class LoftService(Base):
items_csv_reader = csv_helper.file_stream_reader(items_csv)
# add items to attributes dict
attributes['items'] = service_helper.items_csv_to_dict(items_csv_reader, attributes['irt_model'])
attributes['items'] = service_helper.items_csv_to_dict(items_csv_reader)
logging.info('Processed Attributes...')
return attributes