def build_data()

in src/gen_annotation_data.py [0:0]

• 12 lines of code
• 5 McCabe index (conditional complexity)

    def build_data(self, num_tests_per_user: int = 40, archetypes=None):
        data_list = []
        if archetypes is None:
            archetypes = self.all_archetypes
        for arch in archetypes:
            print(f"working on archetype {arch.user_description}")
            for i in range(num_tests_per_user):
                try:
                    data_list.append(self.gen_data_for_archetype(arch))
                except Exception as ex:
                    print(f"Failed instance {i} archetype {arch}: {ex}")
        return pd.concat(data_list, axis=0)