def add_algorithms()

in ml/registry.py [0:0]

• 23 lines of code
• 3 McCabe index (conditional complexity)

    def add_algorithms(self,
                       attrs=[{
                           "classifier": None,
                           "description": None,
                           "status": None,
                           "version": None,
                           "dataset": None,
                           "region": None,
                           "created_by": None
                       }]):

        for attr in attrs:

            if not has_empty_values(attr):
                #get dataset
                dataset, _ = Dataset.objects.get_or_create(
                    name=attr['dataset'], region=attr['region'])

                # get algorithm
                algorithm, _ = Algorithm.objects.get_or_create(
                    classifier=attr['classifier'].__class__.__name__,
                    description=attr['description'],
                    version=attr['version'],
                    status=attr['status'],
                    dataset=dataset,
                    created_by=attr['created_by'])
                self.classifiers[algorithm.id] = attr['classifier']

        return self.classifiers