even more logging

This commit is contained in:
Joshua Burman 2022-02-09 17:25:23 -05:00
parent 3f7b3cc0b6
commit 78331577dc

View File

@ -1,6 +1,8 @@
from pydantic import BaseModel
from typing import List, Optional
import logging
from models.item import Item
from models.constraint import Constraint
from models.irt_model import IRTModel
@ -31,6 +33,7 @@ class SolverRun(BaseModel):
return True
def generate_bundles(self):
logging.info('Generating Bundles...')
bundle_constraints = (constraint.reference_attribute for constraint in self.constraints if constraint.reference_attribute.type == 'bundle')
for bundle_constraint in bundle_constraints:
@ -64,5 +67,7 @@ class SolverRun(BaseModel):
type=type_attribute
)]
logging.info('Bundles Generated...')
def get_constraint(self, name):
return next((constraint for constraint in self.constraints if constraint.reference_attribute.id == name), None)