diff --git a/cli.py b/cli.py index be78e0b..83567a7 100755 --- a/cli.py +++ b/cli.py @@ -41,11 +41,10 @@ class Cli: self.args = ap.parse_args() def calculate_ratings(self): - ratings = [] with Pool(processes=cpu_count()) as pool: - for image in self.args.images: - ratings.append((image, pool.apply_async(self.calculate_rating, - (image, self.args.method)))) + ratings = [(image, pool.apply_async(self.calculate_rating, + (image, self.args.method))) + for image in self.args.images] self.image_ratings = {image: promise.get() for image, promise in ratings} @staticmethod